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

 

---

내 풀이

# 35116kb 76ms

N이 10000만 인 것을 감안해 666+XXXX 4개 자리에 대해 0-9까지 반복했다. 

중복을 포함해서

666 + 숫자 1개 : 2(숫자+666, 666+숫자) * 10(숫자1의 경우의수: 0~9) = 20

666 + 숫자 2개 : 3(숫자1+숫자2+666, 숫자1+666+숫자2, 666+숫자1+숫자2) * 10(숫자1:0~9) *10(숫자2:0~9) = 300

같은식으로 쭉쭉 내려가면

666 + 숫자 4개 : 5 * 10 * 10 * 10 * 10 으로 중복 포함 50000개이므로 N을 충분히 커버할 수 있다고 생각했다. 

 

숫자가 4개이다보니 for문을 4개를 돌리고 반복문은 N번 돌 수 있게 cnt를 세서 break했고,

이 떄 for문을 4번을 돌리다보니break문은 같은 조건으로 3개가 반복되야 했다. 

 

그렇게 입력한 end_list에 숫자를 중복 제거, 순서 정렬 처리하여 입력받은 idx_loc 번째 수가 무엇인지 출력했다. 

idx_loc = int(input())
end = '666'
end_list = []
cnt = 1

for t in range(10):
  for k in range(10):
    for j in range(10):
      for i in range(10):
        t, k, i, j = str(t),str(k),str(i),str(j)

        num1 = int(t + k + j + i + end)
        num2 = int(t + k + j + end + i)
        num3 = int(t + k + end + j + i)
        num4 = int(t + end + k + j + i)
        num5 = int(end + t + k + j + i)
        end_list.extend([num1, num2, num3, num4, num5])

        cnt += 1

      if cnt >= idx_loc:
          break
    if cnt >= idx_loc:
          break
  if cnt >= idx_loc:
          break

print(sorted(set(end_list))[idx_loc-1])

 

메모

모든 수를 돌며 666이 들어간 수를 찾아가며 답을 찾을 수 있지만 (if '666' in Num 같이)

그 경우보다는 더 빨라서 그런가 다른 스터디원들보다는 빠르게 코드가 도는 것 같다.