문제
https://www.acmicpc.net/problem/2839
알고리즘 분류
수학
다이나믹 프로그래밍
그리디 알고리즘
문제 설명
상근이는 요즘 설탕공장에서 설탕을 배달하고 있다. 상근이는 지금 사탕가게에 설탕을 정확하게 N킬로그램을 배달해야 한다. 설탕공장에서 만드는 설탕은 봉지에 담겨져 있다. 봉지는 3킬로그램 봉지와 5킬로그램 봉지가 있다.
상근이는 귀찮기 때문에, 최대한 적은 봉지를 들고 가려고 한다. 예를 들어, 18킬로그램 설탕을 배달해야 할 때, 3킬로그램 봉지 6개를 가져가도 되지만, 5킬로그램 3개와 3킬로그램 1개를 배달하면, 더 적은 개수의 봉지를 배달할 수 있다.
상근이가 설탕을 정확하게 N킬로그램 배달해야 할 때, 봉지 몇 개를 가져가면 되는지 그 수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 N이 주어진다. (3 ≤ N ≤ 5000)
출력
상근이가 배달하는 봉지의 최소 개수를 출력한다. 만약, 정확하게 N킬로그램을 만들 수 없다면 -1을 출력한다.
예시
풀이
5kg으로 먼저 설탕을 묶은 다음, 남은 설탕의 무게 만큼 5kg의 설탕 포대를 해체하여 맞춰준다. 단, 이 때는 특정 상황에서는 풀 설탕 포대가 없을 것이므로 이 때는 -1을 출력하게 한다.
소스코드 1
l=[0,2,4,1,3]
n = int(input())
i = l[n % 5]
j = (n - 3*i) // 5
if j >= 0:
t = i + j
else:
t = -1
print(t)
이 코드의 경우에는, 5kg으로 포장했을 때 남는 kg에 따라 몇 개의 3kg 포대가 더 필요한지 미리 리스트로 기록해둔다. i에 이 리스트를 이용해 사용할 3kg의 포대 수를 구한다. 그 후 j에 남은 설탕으로 5kg 포대 수를 구한다.
하지만 이 때 j의 값이 음수가 될 수 있다. 이 경우는 바로 두 종류의 설탕 포대로는 정확하게 N kg을 만들 수 없음을 의미한다. 그러므로 이 경우에는 -1을 출력하게 한다.
소스코드 2
n=int(input())
bag=0
while n>=0:
if n%5 ==0:
t+=(n//5)
print(t)
break
n-=3
t+=1
else:
print(-1)
이 경우에는 그리디 알고리즘을 이용했다. n을 5로 나눌 수 있는 만큼 최대한 5kg 봉지를 사용하고, 남은 무게를 3kg 봉지로 채우게 했다. 그리고 최종적으로 그게 불가능한 경우, -1을 출력하게 했다.
'Koala - 11기 > 기초 알고리즘 스터디' 카테고리의 다른 글
[백준/Python3] 1330번 : 두 수 비교하기 (0) | 2023.07.16 |
---|---|
[백준/Python] 1551번: 수열의 변화 (0) | 2023.07.15 |
[백준/python] 1267번: 핸드폰 요금 (0) | 2023.07.15 |
[백준/Python] 4458번: 첫 글자를 대문자로 (0) | 2023.07.15 |
[백준/python] 9012번: 괄호 (0) | 2023.07.14 |