Chapter 8-2 1로 만들기

< 이 글은 책 '이것이 취업을 위한 코딩 테스트다' 에서 발췌한 내용을 인용했습니다. >

 

1로 만들기

 

<무지성 코드>

n = int(input())
d = [-1] * 30000
d[1] = 0
i = 2

while i <= n:
    m = 9999999
    if i % 5 == 0 and m > d[i//5] + 1:
        m = d[i//5] + 1
    if i % 3 == 0 and m > d[i//3] + 1:
        m = d[i//3] + 1
    if i % 2 == 0 and m > d[i//2] + 1:
        m = d[i//2] + 1
    if m > d[i-1] + 1:
        m = d[i-1] + 1
    d[i] = m
    i += 1

print(d[n])

<답안 예시>

x = int(input())

d = [0] * 30000

for i in range(2, x+1):
    d[i] = d[i-1] + 1
    if i % 2 == 0:
        d[i] = min(d[i], d[i//2]+1)
    if i % 3 == 0:
        d[i] = min(d[i], d[i//3]+1)
    if i % 5 == 0:
        d[i] = min(d[i], d[i//5]+1)

print(d[x])

 

답안 예시를 보니 while문 보다 for문을 사용하는게 훨씬 깔끔해보인다. 그리고 굳이 하지 않아도 될 변수들을 만들었다. 코드를 다 짠 이후에도 깔끔하게 바꿀 수 있는지 마지막으로 한번 더 생각하는 습관을 들여야 겠다.