[Algorithm] [Python&Swift] 백준/BOJ - 2839 _ 설탕 배달

2839 - 설탕 배달

 

문제

---

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

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

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

 

입력

---

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

 

출력

---

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

 

제출 코드

---

Swift

var sugar = Int(readLine()!)!
var count = 0

while true {
    if sugar % 5 == 0 {
        count += Int(sugar/5)
        print(count)
        break
    }
    sugar -= 3
    count += 1
    if sugar < 0 {
        print("-1")
        break
    }
}

 

Python

sugar = int(input())
count = 0

while True:
    if sugar % 5 == 0:
        count += (sugar // 5)
        print(count)
        break
    sugar -= 3
    count += 1
    if sugar < 0:
        print(-1)
        break

 

문제 풀이

---

전형적인 그리디 문제이다.

N키로그램의 설탕을 5kg, 3kg 봉지를 이용하여 나눠담을때, 최대한 적은 봉지 개수로 나눠담기 위해서는 당연히 5kg을 최대한 사용해야 될 것이다.

따라서 아래와 같은 흐름으로 최소 봉지 개수를 구해낸다.

1. 5kg으로 나눠질 경우, 남은 설탕을 모두 5kg 봉지에 나눠담고 봉지 개수를 출력한다.
2. 그렇지 않을 경우, 3kg 봉지를 하나 사용한다.
3. 만일 설탕이 음수일 경우, 5kg와 3kg 봉지에 맞춰 나눠 담을 수 없는 수 임으로 -1을 출력한다.