[백준] 1436: 영화감독 숌 (Python)

1436: 영화감독 숌

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

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번째 영화의 제목에 들어간 수를 출력한다.

풀이

 10000번째 숫자가 6669999보다 작으므로 브루트 포스로 연산할 수 있다. 따라서 666부터 6669999까지 차례대로 숫자에 666이 있는지 검사하고 n번째 숫자를 출력한다. 내가 처음 작성한 코드는 다음과 같다. 

# 브루트 포스 - 1436: 영화감독 숌
n = int(input())
arr = [0] * (n+1)
index = 1
for i in range(666, 6669999):
    num = str(i)
    if num.find("666") == -1:
        continue
    else:
        arr[index] = i
        index += 1
    if index > n:
        break
print(arr[n])

 정답 판정을 받았으나 다시 생각해보니 다듬을 수 있는 부분이 몇 군데 있었다. 우선 dp 문제를 풀던 습관이 남아있어서인지 자꾸 리스트에 구한 값을 저장하려고 하는데 그러지 않고 n번째 숫자를 찾았을 때 바로 출력해도 되므로 arr 리스트를 없앴다. 리스트를 없앴으므로 index 변수의 이름은 count로 수정했다. 그리고 굳이 num이라는 변수에 str(i)를 저장할 필요 없이 바로 str(i)에 find() 함수를 적용해도 되므로 num 변수를 없앴다. 이렇게 수정한 코드는 다음과 같다.

n = int(input())
count = 1
for i in range(666, 6669999):
    if str(i).find("666") == -1:
        continue
    else:
        count += 1
    if count > n:
        print(i)
        break

 이후 다른 사람들의 풀이를 찾아보니 로직은 동일하나 더 간결한 코드가 많았다. for문 대신 while문을 돌리고 while문을 while(n)으로 작성한 후 n--를 하는 방법도 있었고, while(1) 안에서 666을 저장한 변수 result를 두고 result += 1을 하며 하나씩 검사하는 방법도 있었다. for문만 쓰지 말고 while문도 잘 활용할 수 있도록 해야겠다. 

 참고로 나는 find함수를 이용해 substring을 검사했지만 if '666' in str(result): 와 같이 검사하는 방법도 있다.