Submission #48548762

---

Source Code Expand

import os
import sys
import operator  # noqa
from typing import (IO, Any, Callable, Dict, Iterable, Iterator, List,  # noqa
                    Optional, Set, Tuple, Union)

sys.setrecursionlimit(210000)


if os.getenv('TEST'):
    def eprint(*args, **kwargs):
        print('[EPRINT]', *args, file=sys.stderr, **kwargs)
else:
    def eprint(*args, **kwargs):
        pass


def main() -> None:
    N = int(input())
    nums = [0] * 3
    for _ in range(N - 1):
        if nums[0] == nums[2]:
            nums[0] = 0
            nums[1] = 0
            nums[2] += 1
        elif nums[0] == nums[1]:
            nums[0] = 0
            nums[1] += 1
        else:
            nums[0] += 1
        eprint(nums)

    bases = [1]
    for _ in range(nums[-1]):
        bases.append(bases[-1] * 10 + 1)

    print(sum(bases[n] for n in nums))


if __name__ == '__main__':
    main()

Submission Info

Judge Result