설탕 배달

박고은·2023년 7월 14일
0

코딩테스트 연습

목록 보기
29/34

문제 ✨

2839번: 설탕 배달

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

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

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

N = int(input())

sugar = []

for i in range(0, N//5+1):
    for j in range(0, N//3+1):
        if i*5+j*3==N: sugar.append(i+j)

if not sugar: print(-1)
else: print(min(sugar))

동적 계획법으로 풀이한 코드

N = int(input())

dp = [5001]*(N+1)
dp[3] = 1

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

print(-1) if dp[N]>5000 else print(dp[N])

3킬로그램 봉지 추가와 5킬로그램 봉지 추가 중에 전체 봉지 수가 더 적은 경우를 선택해서 +1
최솟값으로 봉지 갯수를 계산하므로 N의 최댓값인 5000보다 큰 5001로 dp 배열 구성

N = 11의 경우, dp = [5001, 5001, 5001, 1, 5001, 1, 2, 5002, 2, 3, 2, 3]
dp[7]은 i-3인 dp[4]와 i-5인 dp[2]가 5001으로 5002가 되어 최솟값으로 반환되지 않음 -> 봉지 갯수 계산 과정에서 제외

0개의 댓글