백준 1436번 파이썬

1436번: 영화감독 숌

처음에 문제를 읽어 의미를 파악하지 못했다. 말그대로, 입력받은 n에 대해, 666이 들어간 모든 숫자 중 n번째 인 수를 구하는 것이다. n 은 10,000 이하이므로, _ 666 _ _ _=> 9 * 10 * 10 * 10 : 9천 가지수 가능. 따라서 나는 어림잡아 666부터 9666999까지 만약 666이 숫자 내 들어있다면 빈 배열에 추가하여 n 번째 수를 찾는 방식을 썼다. 

하지만 이 방식은 모든 경우를 찾기 때문에 시간 초과가 날 수 있으므로 count 변수를 사용해서 n 번째에 도달했을 때 break를 해줬다.

n = int(input())
res = [0]
count = 0
for i in range(666, 9666999):
    target  = str(i)
    if '666' in target:
        count += 1
        res.append(i)
        if count == n:
            break
print(res[n])