백준 2839번. 설탕배달 (Python / 파이썬)

문제

상근이는 요즘 설탕공장에서 설탕을 배달하고 있다. 상근이는 지금 사탕가게에 설탕을 정확하게 N킬로그램을 배달해야 한다. 설탕공장에서 만드는 설탕은 봉지에 담겨져 있다. 봉지는 3킬로그램 봉지와 5킬로그램 봉지가 있다.
상근이는 귀찮기 때문에, 최대한 적은 봉지를 들고 가려고 한다. 예를 들어, 18킬로그램 설탕을 배달해야 할 때, 3킬로그램 봉지 6개를 가져가도 되지만, 5킬로그램 3개와 3킬로그램 1개를 배달하면, 더 적은 개수의 봉지를 배달할 수 있다.
상근이가 설탕을 정확하게 N킬로그램 배달해야 할 때, 봉지 몇 개를 가져가면 되는지 그 수를 구하는 프로그램을 작성하시오.

입력 

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

출력

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

---

문제 풀이

설탕 배달 문제는 그리디 문제입니다. 그리디는 가장 좋아보이는 것을 우선적으로 선택하는 알고리즘입니다. 

총 N 킬로그램의 설탕을 3 킬로그램, 5 킬로그램 봉지로만 배달 가능하며 최소한의 봉지만 사용해야 합니다.

따라서 이문제의 아이디어는 가능한 최소한의 봉지를 사용해야 하기 때문에

5 킬로그램 봉지를 최대한 많이 사용하면 좋습니다. 

 

따라서 설탕의 무게 N을 5의 배수로 만들어주어야 합니다. 

5의 배수가 될 때 까지 N에서 3을 빼주고, 이는 3 킬로그램 봉지를 사용한 것이므로

빼줄 때마다 봉지 수를 count 해줍니다.

N이 5의 배수가 되면 N을 5로 나눈 몫을 봉지 수로 count 해줍니다.

 

이때, 주의해야 할 점이 있습니다. 숫자 4처럼 설탕배달을 할 수 없는 예외 무게도 존재합니다.

따라서 예외처리를 꼭 꼭!! 해주어야 합니다.

---

My code

n = int(input()) # 설탕의 무게 입력

count = 0 # 봉지 수를 count
while n > 0:
    if n % 5 == 0: #5의 배수가 되면
        count += n // 5 # 5로 나눈 몫을 count에 더해줌
        break
    n -= 3 # 5의 배수가 될 때까지 3을 빼줌
    count += 1 # 3 kilo 봉지 사용했으므로 count
    
if n >= 0:
    print(count) 
else:
    print(-1)   # n이 음수이면 예외처리