[백준 1436번] 영화감독 숌

처음 짠 코드

N = int(input())

i = 0

j = 666

def check_666(j):

  J = str(j)

  for i in range(len(J)-2):

    if J[i]=='6':

      if J[i+1]=='6':

        if J[i+2]=='6':

          return True

 

  return False

 

while N>i:

  ans = check_666(j)

  if ans==True:

    i+=1

    j+=1

  else:

    j+=1

print(j-1)

 

개선된 코드

N = int(input())

i = 0

j = 666

def check_666(j):

  return '666' in str(j)

 

while N > i:

  if check_666(j):

    i += 1

  if N > i:  # N번째 영화를 찾기 전까지만 j를 증가

    j += 1

print(j)  # 이미 조건을 만족하는 j 값을 찾았으므로 j-1 없이 j 출력