[Python]2839번, 설탕 배달

목차

 

문제

 

상근이는 요즘 설탕공장에서 설탕을 배달하고 있다. 

상근이는 지금 사탕가게에 설탕을 정확하게 N킬로그램을 배달해야 한다. 

설탕공장에서 만드는 설탕은 봉지에 담겨 있다. 봉지는 3킬로그램 봉지와 5킬로그램 봉지가 있다.

상근이는 귀찮기 때문에, 최대한 적은 봉지를 들고 가려고 한다. 

예를 들어, 18킬로그램 설탕을 배달해야 할 때, 3킬로그램 봉지 6개를 가져가도 되지만, 

5킬로그램 3개와 3킬로그램 1개를 배달하면, 더 적은 개수의 봉지를 배달할 수 있다.

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

 

입력

 

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

 

출력

 

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

 

풀이

 

오랜만의 DP 문제이다.

 

취업하고 쫌쫌따리 알고리즘을 풀고 있긴 하지만 기억에 잘 남는 것 같지 않아서,

 

아예 종목을 정해 한 가지 유형만 패보기로 했다.

 

문제는 간단하다. 주어진 무게를 3킬로와 5킬로 포대를 합쳐 최소 개수로 배달하는 것.

 

주어지는 숫자는 N 하나이고, 3 이상 5000 이하의 숫자이기 때문에

 

N+1개의 요소를 가지는 리스트 dp를 만들고 모든 숫자를 5001로 채운다.

 

이어서 dp[3]=1로, N이 5 이상일 경우에는 dp[5]=1로 초기화해 주고 동적계획법으로 리스트를 채우면 된다.

 

여기서 N이 3이라면 답은 1, 4라면 -1(불가능하기 때문에), 5라면 1로 답이 정해지며

 

6부터 N+1까지 반복문을 돌며 채운다.

 

채우는 알고리즘은 다음과 같으며,

 

• 만약 dp[i - 3]이 불가능한 값(5001)이 아니라면, dp[i] = min(dp[i], dp[i - 3] + 1)
• 만약 dp[i - 5]이 불가능한 값(5001)이 아니라면, dp[i] = min(dp[i], dp[i - 5] + 1)

둘 중 최솟값을 가져오기 때문에 순서는 중요하지 않다.

 

또한 당연하게도 N이 5보다 작은 경우엔 반복문은 실행되지 않는다.

 

값을 N까지 채워주고 나면, 조건에 따라 dp[N]이나 -1을 출력한다.

 

Python

 

import sys

N = int(sys.stdin.readline().rstrip())
dp = [5001] * (N + 1)
dp[3] = 1

if N >= 5:
    dp[5] = 1

for i in range(6, N + 1):
    if dp[i - 3] != 5001:
        dp[i] = min(dp[i], dp[i - 3] + 1)
    if dp[i - 5] != 5001:
        dp[i] = min(dp[i], dp[i - 5] + 1)

print(dp[N] if dp[N] != 5001 else -1)

풀이를 코드로 옮긴 것이다.

 

N이 3보다는 무조건 크거나 같기 때문에 조건에 상관없이 초기화를 1로 시켜주어야 한다.

 

마지막 줄에서는 삼항 연산자를 사용해 보았다.

 

앞으로의 DP 글은 별다른 일이 없다면 파이썬으로만 진행할 생각이다.

 

간결하고, 보기 좋고, 이해하기도 편한 것 같아서.

 

백준 2839번 설탕 배달, 끝!