[백준] 2839 설탕배달

iamjinseo·2022년 7월 14일

문제풀이-Python

목록 보기

28/134

문제

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

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

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

입출력

입력

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

출력

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

입출력 예시

해설

문제 푼 사람 수가 많을 정도로 유명한 문제인데, 난 그리디 문제인줄로만 알고 시도했다가 dp로도 풀 수 있길래 둘 다 시도해봄.

그리디

이코테 1이 될 때까지 문제랑 비슷하다. 그 문제에서는 최대한 나누기를 시도해야 시도 횟수가 적어진다.

이 문제도 그와 비슷하게 최대한 5로 나누어야 봉지의 갯수가 최소가 된다.
N이 주어졌을 때 N이 5로 나누어진다면 나누어서 몫을 출력하고, 그렇지 않다면 N이 5의 배수가 될 때 까지 3을 계속 빼면 된다.

코드

# N이 5의 배수 될 때 까지 3계속 빼기
import sys
N = int(sys.stdin.readline())

count = 0
while(True):
    if N % 5 == 0:
        count+=N//5        
        break
    else:
        # N이 3보다 작으면 안됨
        if N < 3:
            count = -1
        N -= 3 
        count +=1
print(count)

생각

if N < 3:
     count = -1

이거 N-=3 count+=1 밑에 놨었는데 그렇게 하면 오류남

다이나믹 프로그래밍

그림에서 위의 숫자는 N이고 아래 숫자는 봉지 개수라고 보면 된다.
15를 보면 왜 봉지 개수가 3이라는 결과가 나왔는지에 대한 해설이 빨간 글씨로 적혀 있다.

요약하자면 N이 주어졌을 때 N-3kg의 개수와 N-5kg의 개수를 비교해 더 작은 쪽에서 +1을 하면 됨(+3kg 또는 +5kg니까)

코드

import sys
N = int(sys.stdin.readline())

# dp 로 풀기
# N이 되기까지의 과정을 메모이제이션
dp = [0]*(5001)
dp[1] = -1
dp[2] = -1
dp[3] =1
dp[4] = -1
dp[5] = 1

for i in range(6, 5001):
    # i에서 3을 뺀 수의 경우와 5를 뺀 수의 경우의 수중 더 낮은 걸 골라(더 적은 봉지 수)
    # +3kg 을 하든 +5kg 를 하든 하면 i일 때의 봉지의 수는 이전 수의 봉지의 수에서 +1이 되므로

    # 어느 한쪽이라도 -1이라면 다른 쪽의 봉지의 수에서 +1
    if dp[i-3] == -1 :
        dp[i] = dp[i-5]+1
    elif dp[i-5] == -1:
        dp[i] = dp[i-3]+1
    elif dp[i-3]!=-1 and dp[i-5]!=-1: #어느 한쪽도 -1이 아닐 때
        dp[i] = min(dp[i-3], dp[i-5])+1
    
    if dp[i] == 0 :
        dp[i] = -1 # 만약 둘다 -1이면 0이라는 값이 나오므로 그냥 -1로 만들어버리기
print(dp[N])