[백준] 1436 - 영화감독 숌

문제 ( www.acmicpc.net/problem/1436 )

666은 종말을 나타내는 숫자라고 한다. 따라서, 많은 블록버스터 영화에서는 666이 들어간 제목을 많이 사용한다. 영화감독 숌은 세상의 종말 이라는 시리즈 영화의 감독이다. 조지 루카스는 스타워즈를 만들 때, 스타워즈 1, 스타워즈 2, 스타워즈 3, 스타워즈 4, 스타워즈 5, 스타워즈 6과 같이 이름을 지었고, 피터 잭슨은 반지의 제왕을 만들 때, 반지의 제왕 1, 반지의 제왕 2, 반지의 제왕 3과 같이 영화 제목을 지었다.

하지만 숌은 자신이 조지 루카스와 피터 잭슨을 뛰어넘는다는 것을 보여주기 위해서 영화 제목을 좀 다르게 만들기로 했다.

종말의 숫자란 어떤 수에 6이 적어도 3개이상 연속으로 들어가는 수를 말한다. 제일 작은 종말의 숫자는 666이고, 그 다음으로 큰 수는 1666, 2666, 3666, .... 과 같다.

따라서, 숌은 첫 번째 영화의 제목은 세상의 종말 666, 두 번째 영화의 제목은 세상의 종말 1666 이렇게 이름을 지을 것이다. 일반화해서 생각하면, N번째 영화의 제목은 세상의 종말 (N번째로 작은 종말의 숫자) 와 같다.

숌이 만든 N번째 영화의 제목에 들어간 숫자를 출력하는 프로그램을 작성하시오. 숌은 이 시리즈를 항상 차례대로 만들고, 다른 영화는 만들지 않는다.

 

입력

첫째 줄에 숫자 N이 주어진다. N은 10,000보다 작거나 같은 자연수이다.

 

출력

첫째 줄에 N번째 영화의 제목에 들어간 수를 출력한다.

 

예제

2

1666

 

나의 코드

N = int(input())

returnValue = 666 # start with first value(includes '666')
count = 0
while True:
    if '666' in str(returnValue): # check current value includes '666'
        count += 1
    if count == N: break
    returnValue += 1

print(returnValue)

• 666이 위치 관계없이 숫자 안에 존재만 하면 되므로, String형으로 비교한다.
• '666'이 있는 경우에만 count+=1를 수행하여 666을 포함한 몇번째 숫자인지 찾는다.
• 666(첫번째 666을 포함한 숫자) 부터 시작해서 1씩 증가시켜가면서 모든 숫자를 검사한다. (brute force)

 

+ 초기 코드 - 시간 초과

strN = input()
N = int(strN)

returnValue = 665
count = 0
while True:
    if count == N: break

    returnValue+=1
    for i in range(len(strN)): # check if current value includes '666'
        if (returnValue / (10 ** i))%1000 == 666:
            count += 1
            break

print(returnValue)

• 666을 integer로 비교하기 위한 이중 반복문 사용으로 시간 초과가 발생하여 구조를 바꿈