[Algorithm] 백준 2839

문제

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

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

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

풀이

내가 생각한 알고리즘 : 오답

1. 전체 kg을 5로 나눈 것의 몫을 5kg 봉투 갯수로 생각
2. 그 나머지를 3으로 나눈 것의 몫을 3kg 봉투 갯수로 생각
3. 만약 나머지가 0이 아니면 -1 출력

이렇게 알고리즘을 짰는데, 9 의 경우 예외가 발생한다.
9 / 5 = 1
4 / 3 = 1
나머지 1

하지만
9 / 3 = 3
나머지 0

오답코드

n = int(input())

a, b = divmod(n, 5)
b, c = divmod(b, 3)

result = -1
if c == 0:
    result = a + b
print(result)

내가 생각한 알고리즘 : 정답

가능한 모든 경우의 수를 확인한 후, min 값을 가져가는 것을 생각했다.

1. n을 입력 받는다.
2. 5kg 봉투의 최대값을 구한다. : 반복 횟수로 사용한다.
3. 모든 경우의 수에 대해 계산 후, 나머지가 0 이라면 최솟값을 갱신한다.

정답코드

n = int(input())

a = int(n/5)

result = n

cnt_3 = 0
i = 0

for i in range(0, a + 1):
    cnt_3, m = divmod(n-(5*i), 3)
    if m == 0:
        result = min(result, cnt_3 + i)

if result == n:
    result = -1
print(result)

Lesson Learn

모든 경우의 수에 대해 생각하기.
그리고 손코딩할때, 작은 수 말고 큰 수를 넣어보고 일반화를 하자 !
DP로도 풀 수 있겠다. 3 씩, 5 씩 커지는 로직을 생각하면 될거 같다.

divmod, min 내장함수를 적절히 잘 사용한거 같당. :)