[Baekjoon] 2839 : 설탕 배달 - python

문제

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

 

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

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

 

▶️ 입력 조건

• 첫째 줄에 N이 주어진다. (3 ≤ N ≤ 5000)

▶️ 출력 조건

• 상근이가 배달하는 봉지의 최소 개수를 출력한다. 만약, 정확하게 N킬로그램을 만들 수 없다면 -1을 출력한다.

---

풀이

해당 문제는 거스름 돈과 거의 같다고 생각해서 단순한 탐욕적인 접근 법으로 5kg으로 먼저 나누고 시작하면 된다고 생각했는데,

경우의 수를 보니 단순하게 5kg로만 나누고 시작해서 될 문제는 아닌 것 같았다.

문제를 살펴보면 조건은 총 4가지이다.

 

1) 5kg으로 나누어 떨어짐

2) 3kg으로 나누어 떨어짐

3) 최대한의 5kg과 최소한의 3kg을 혼합

4) 3가지 모두 충족하지 않을 경우 -1 

 

1번 2번 4번은 단순한 조건문이므로 문제가 되지 않는다. 2번을 해결하기 위해 생각을 해보았는데,

본인이 고안한 방법은 "n에서 -3을 하고 5로 나누어 떨어지는지 검증" => 이 과정을 반복하는 것이였다.

따라서 해당 방법을 코드로 구현함으로써 해결할 수 있었다.

 

n = int(input())

def get_sugar_count(n):
    
    bag_3 = 0
    
    while True:
        # 정확하게 나누어 떨어지지 않는 경우
        if n == 1 or n == 2:
            return -1
        
        # 3kg 봉지로만 나눌 수 있는 경우
        elif n == 0:
            return bag_3
        
        # 5kg 봉지로만 혹은 최대한의 5kg와 최소한의 3kg 봉지로 나뉘는 경우
        else:
            if n % 5 == 0:
                bag_5 = n // 5
                return bag_3 + bag_5
            n -= 3
            bag_3 += 1

print(get_sugar_count(n))