문제
문제 링크
상근이는 요즘 설탕공장에서 설탕을 배달하고 있다. 상근이는 지금 사탕가게에 설탕을 정확하게 N킬로그램을 배달해야 한다. 설탕공장에서 만드는 설탕은 봉지에 담겨져 있다. 봉지는 3킬로그램 봉지와 5킬로그램 봉지가 있다.
상근이는 귀찮기 때문에, 최대한 적은 봉지를 들고 가려고 한다. 예를 들어, 18킬로그램 설탕을 배달해야 할 때, 3킬로그램 봉지 6개를 가져가도 되지만, 5킬로그램 3개와 3킬로그램 1개를 배달하면, 더 적은 개수의 봉지를 배달할 수 있다.
상근이가 설탕을 정확하게 N킬로그램 배달해야 할 때, 봉지 몇 개를 가져가면 되는지 그 수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 N이 주어진다. (3 ≤ N ≤ 5000)
출력
상근이가 배달하는 봉지의 최소 개수를 출력한다. 만약, 정확하게 N킬로그램을 만들 수 없다면 -1을 출력한다.
Code
import sys
input = sys.stdin.readline
N = int(input().rstrip())
list_ = [-1 for _ in range(5001)]
list_[3] = 1
list_[5] = 1
for i in range(6, N+1) :
if list_[i-3] < 0 and list_[i-5] < 0:
list_[i] = -1
elif list_[i-3] > 0 and list_[i-5] > 0 :
list_[i] = min(list_[i-3], list_[i-5]) + 1
else :
list_[i] = max(list_[i-3], list_[i-5]) + 1
print(list_[N])코드 설명
import sys
input = sys.stdin.readline
N = int(input().rstrip())
list_ = [-1 for _ in range(5001)]
list_[3] = 1
list_[5] = 1빠른 입력을 받을 수 있도록 input을 선언하고 설탕의 무게를 받는다. 문제에서 N의 최댓값은 5000이라 했으므로 -1이 5001개(0번 인덱스 고려) 들어있는 리스트를 생성한다. 3kg과 5kg일 경우 봉지 1개씩만 들고가면 되며, 이 부분이 DP의 시작점이므로 두 개는 -1이 아닌 1로 설정한다.
for i in range(6, N+1) :
if list_[i-3] < 0 and list_[i-5] < 0:
list_[i] = -1
elif list_[i-3] > 0 and list_[i-5] > 0 :
list_[i] = min(list_[i-3], list_[i-5]) + 1
else :
list_[i] = max(list_[i-3], list_[i-5]) + 1
print(list_[N])for문을 돌려서 i보다 3 작은 수와 5 작은 수 모두 -1일 경우 가능한 경우의 수가 없다는 뜻이므로 i번째 인덱스에 -1을 넣는다. 만약 3 작은 수와 5 작은 수 모두 -1이 아닐 경우에는 가능한 경우의 수가 두 개라는 의미이므로 둘 중 작은 수 + 1을 한다. +1을 하는 이유는 설탕 봉지를 하나 더 챙긴다는 의미이다. 만약 둘 중에 하나만 -1이 아닌 경우에는 -1을 피해야 하므로 max값 +1을 한다.
리뷰
미루고 미루던 설탕 배달... if문 쓸 때 말도 안 되는 실수하긴 했는데 생각보다 오래 안 걸려서 괜히 여태 미뤘다는 생각을 했다. 오히려 스택 할 때보다 DP 할 때가 더 재밌는 거 같다. 아ㅏㅓㄹ우ㅠㅣㅏㅠㅠㅠㅠㅠ 쓰던 거 날라가서 다시 써야 돼
