[Python/파이썬] 백준 2839번 설탕 배달

2839번: 설탕 배달

문제

상근이는 요즘 설탕공장에서 설탕을 배달하고 있다. 상근이는 지금 사탕가게에 설탕을 정확하게 N킬로그램을 배달해야 한다. 설탕공장에서 만드는 설탕은 봉지에 담겨져 있다. 봉지는 3킬로그램 봉지와 5킬로그램 봉지가 있다.

상근이는 귀찮기 때문에, 최대한 적은 봉지를 들고 가려고 한다. 예를 들어, 18킬로그램 설탕을 배달해야 할 때, 3킬로그램 봉지 6개를 가져가도 되지만, 5킬로그램 3개와 3킬로그램 1개를 배달하면, 더 적은 개수의 봉지를 배달할 수 있다.

상근이가 설탕을 정확하게 N킬로그램 배달해야 할 때, 봉지 몇 개를 가져가면 되는지 그 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N이 주어진다. (3 ≤ N ≤ 5000)

출력

상근이가 배달하는 봉지의 최소 개수를 출력한다. 만약, 정확하게 N킬로그램을 만들 수 없다면 -1을 출력한다.

코드

n = int(input())
dp = [0, -1, -1, 1, -1, 1]

if n > 5:
    for i in range(6, n + 1):
        if dp[i-3] == -1:   # 3kg 안됨
            if dp[i-5] == -1:   # 만들 수 없는 경우
                dp.append(-1)
            else:               # 3kg X, 5kg O
                dp.append(dp[i-5] + 1)
        else:               # 3kg 됨
            if dp[i-5] == -1:   # 3kg O, 5kg X
                dp.append(dp[i-3] + 1)
            else:               # 3kg O, 5kg O
                dp.append(min(dp[i-3], dp[i-5]) + 1)
print(dp[n])

상근이가 배달하는 봉지의 최소 개수를 저장하는 배열 dp를 만들 것이다. dp[N]에는 N킬로그램일 때의 봉지의 최소 개수가 들어간다.

초기값으로는 [0, -1, -1, 1, -1, 1]을 넣어주었는데 만들 수 없는 값은 -1로 표시하였다.

이제 dp배열에 값을 채워줄 것인데 i킬로그램이 3킬로그램 봉지와 5킬로그램 봉지로 만들 수 있는지 확인하며 추가해야 한다.

1. dp[i-3] == -1, dp[i-5] == -1이라면
   3킬로그램과 5킬로그램 봉지 어떤 것을 갖고 가도 이 무게는 만들 수 없다. 따라서 -1을 저장한다.
2. dp[i-3] == -1, dp[i-5] != -1이라면
   5킬로그램 봉지를 갖고 가야만 한다. 따라서 dp[i-5] + 1을 저장한다.
3. dp[i-3] != -1, dp[i-5] == -1이라면
   3킬로그램 봉지를 갖고 가야만 한다. 따라서 dp[i-3] + 1을 저장한다.
4. dp[i-3] != -1, dp[i-5] != -1이라면
   dp[i-3]과 dp[i-5] 중 작은 값을 골라야 한다. 따라서 min(dp[i-3], dp[i-5]) + 1을 저장한다.

이렇게 진행한 후 dp[n]의 값을 출력해주면 된다.