백준 1436 Python

https://www.acmicpc.net/problem/1436

1436번: 영화감독 숌

시간이 너무 오래걸려서 if elif 문으로 런타임을 줄임

import sys

N = int(sys.stdin.readline())
num = ''
cnt = 0
i = 0
tm = N // 1000

if N > 9000 :
    cnt = 9000
    i = 2566679
elif N > 8000 :
    cnt = 8000
    i = 2221666
elif N > 7000 :
    cnt = 7000
    i = 1866639
elif N > 6000 :
    cnt = 6000
    i = 1666499
elif N > 5000 :
    cnt = 5000
    i = 1466659
elif N > 4000 :
    cnt = 4000
    i = 1110666
elif N > 3000 :
    cnt = 3000
    i = 766619
elif N > 2000 :
    cnt = 2000
    i = 666199
elif N > 1000 :
    cnt = 1000
    i = 366639


while cnt < N:
    i += 1
    num = str(i)
    if "666" in num:
        cnt += 1

print(num)