문제

링크

풀이

문제를 처음 보고 풀었을 때 굉장히 쉬운 문제인 줄 알았다. N=1 일 때만 666을 출력하고, N>1 일 때는 (N-1)666을 출력하면 되는 쉬운 문제로 풀었다. 하지만, 자세히 보니 함정이 있었다...

문제에 나와있듯이 'N번째 영화 제목은 N번째로 작은 종말의 숫자'라는 점을 봐야한다. 666, 1666, 2666, 3666, 4666, 5666, ... 으로 흘러간다. 이 때 다음 7번째 영화 제목은 6666이 아닌 6660 이다! 즉, 6660, 6661, 6662, ... 순서로 흘러간다. 위 포인트를 파악한다면 쉽게 문제를 풀 수 있다.

코드

import sys

N = int(sys.stdin.readline())

# 시간복잡도 조금이라도 줄이기 위해 666부터 시작
cnt, movie = 0, 666
while 1:
    if '666' in str(movie):
        cnt += 1
        if cnt == N:
            print(movie)
            break
    movie += 1