백준 알고리즘 1436 (영화감독 숌) - python (재풀이)

백준 알고리즘 1436 (영화감독 숌) - python

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

1436번: 영화감독 숌

 

666이 들어간 수열을 순서대로 찾는 문제.

str으로 접근하니 간단했다.

 

[문제 풀이]

1. 666에서 시작하여 str 기준으로 먼저 '6'을 발견하면

2. 그 idx부터 3개가 '666'인지는 체크

3. 맞으면 N을 -1하고 체크는 중단

4. 숫자를 올리면서 계속 반복

5. N == 0이 되면 출력

 

import sys
N = int(sys.stdin.readline())
num = 666  # 시작 666
while N > 0:
    for idx, str_num in enumerate(str(num)):
        if str_num == '6':  # 6이 발견되면
            # 666인지 체크
            if str(num)[idx:idx+3] == '666':
                N -= 1
                break
    if N == 0:
        print(num)
        break
    num += 1