[백준🥈3] #1463 1로 만들기 (Python)

문제

정수 X에 사용할 수 있는 연산은 다음과 같이 세 가지 이다.

1. X가 3으로 나누어 떨어지면, 3으로 나눈다.
2. X가 2로 나누어 떨어지면, 2로 나눈다.
3. 1을 뺀다.

정수 N이 주어졌을 때, 위와 같은 연산 세 개를 적절히 사용해서 1을 만들려고 한다. 연산을 사용하는 횟수의 최솟값을 출력하시오.

입력

첫째 줄에 1보다 크거나 같고, 106보다 작거나 같은 정수 N이 주어진다.

출력

첫째 줄에 연산을 하는 횟수의 최솟값을 출력한다.

예제 입력 1

2

예제 출력 1

1

예제 입력 2

10

예제 출력 2

3

 

❌ Code

n = int(input())
dp = [0] * (n+3)

dp[1] = 0
dp[2] = 1
dp[3] = 1

for i in range(4, n+1):
    if i % 3 == 0:
        dp[i] = min(dp[i//3], dp[i-1]) + 1
    
    elif i % 2 == 0:
        dp[i] = min(dp[i//2], dp[i-1]) + 1
    
    else:
        dp[i] = dp[i-1] + 1

    # print(i, '=', dp[i])
    
print(dp[n])

 

✔️ Code

n = int(input())
dp = [0] * (n+1)

dp[1] = 0

for i in range(2, n+1):
    
    dp[i] = dp[i-1] + 1 
    # 1을 빼서 연산한 것을 먼저 넣어주기

    if i % 3 == 0:
        dp[i] = min(dp[i//3] + 1, dp[i])
    
    if i % 2 == 0:
        dp[i] = min(dp[i//2] + 1, dp[i])
    
    # 3으로 나눠질 때와 2로 나눠질 때 모두 비교해봐야 하므로 elif 사용 x

    
print(dp[n])

 

✏️ Comment

처음에 당연히 if문으로 풀면 될 줄 알고 코드를 작성했는데 50% 정도까진 통과했다가 "틀렸습니다"가 나왔다. 뒤의 테스트 케이스 중에서 해당 코드로는 통과되지 않는 부분이 있었던 것 같다. 결론적으로 3으로 나누는 것, 2로 나누는 것, 1을 빼는 것 다 돌려보고 그 중에 최솟값을 찾아야할 것 같았다.

통과한 코드는 구글링을 통해 참고한 코드인데, 우선 1을 빼준 것의 dp 값을 저장해준 뒤, 3으로 나눠지면 3으로 나눈 dp 값에 +1 한 것과 -1을 해준 dp 값을 비교, 2로 나눠지면 2로 나눈 dp 값에 +1 한 것과 -1을 해준 dp 값을 비교하여 최솟값을 저장하는 식으로 구현했다.