상근이는 요즘 설탕공장에서 설탕을 배달하고 있다. 상근이는 지금 사탕가게에 설탕을 정확하게 N킬로그램을 배달해야 한다. 설탕공장에서 만드는 설탕은 봉지에 담겨져 있다. 봉지는 3킬로그램 봉지와 5킬로그램 봉지가 있다.
상근이는 귀찮기 때문에, 최대한 적은 봉지를 들고 가려고 한다. 예를 들어, 18킬로그램 설탕을 배달해야 할 때, 3킬로그램 봉지 6개를 가져가도 되지만, 5킬로그램 3개와 3킬로그램 1개를 배달하면, 더 적은 개수의 봉지를 배달할 수 있다.
상근이가 설탕을 정확하게 N킬로그램 배달해야 할 때, 봉지 몇 개를 가져가면 되는지 그 수를 구하는 프로그램을 작성하시오.
첫째 줄에 N이 주어진다. (3 ≤ N ≤ 5000)
상근이가 배달하는 봉지의 최소 개수를 출력한다. 만약, 정확하게 N킬로그램을 만들 수 없다면 -1을 출력한다.
18
4
4
-1
6
2
9
3
11
3
import sys
input = sys.stdin.readline
N = int(input())
dp=[]
answer = []
count = 0
if N % 5 == 0:
answer.append(N // 5)
elif N % 3 == 0:
answer.append(N // 3)
for i in range(1, (N // 5) + 1):
dp.append([i,N - (i*5)]) # dp[0] == i, dp[1] == 남은 수들
for j in range(len(dp)):
temp = dp[j][1] % 3
if temp == 0:
count = dp[j][0] + dp[j][1] // 3
answer.append(count)
if len(answer) == 0:
answer.append(-1)
print(min(answer))
별거 아닌거 같아보이는데 좀 깊게 생각해봐야 할 edge case들이 있던 문제였다.
우선 설탕 양이 5의 배수이거나 3의 배수인 경우에 적용할 수 있도록 answer 리스트에 그 값을 넣어준다.
그리고 예제의 18이나 11처럼 5kg 봉투와 3kg 봉투를 섞어서 써야 최소값이 나오는 경우를 계산하기 위해서 우선 설탕양에서 5의 배수들을 뺀 값을 dp 배열에 기록했다.
그리고 그 남은 양들을 3으로 나눴을 때 나머지가 0이라면 설탕 봉투를 몇개나 써야하는지 알 수 있으니까 answer 리스트에 append 한다.
이와 같은 절차를 거쳤어도 answer 리스트에 아무것도 넣지 못한 수라면 애초에 5와 3으로 딱 떨어지지 않는 수라는 뜻이므로 answer에 -1을 넣어준다.
그리고 answer 리스트의 최소값을 출력하면 해결!
DP와 그리드를 한번에 적용해야 하는 문제였다. 이렇게 푸는게 맞나 싶긴 하지만.