[백준] Swift - 2839번: 설탕 배달

👆문제풀러 가기👆

문제

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

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

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

입력

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

출력

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

알고리즘 분류

• 수학
• 다이나믹 프로그래밍
• 그리디 알고리즘

---

DP 풀이 [메모리: 69100KB,  시간: 16ms]

let sugarWeight = Int(readLine()!)! // 설탕무게
var dp = Array(repeating: 5001, count: 5001) // 설탕 최대무게 = 5000
dp[3] = 1 // 설탕 3kg일 때 1개
dp[5] = 1 // 설탕 5kg일 때 1개

for i in 6...5000 {
    // 6kg이면 dp[3]과 dp[1] 비교
    // 8kg이면 dp[5]와 dp[3] 비교
    // 11kg이면 dp[8]과 dp[6] 비교
    dp[i] = min(dp[i-3], dp[i-5]) + 1
}
// 정상적인 값이 반영되지 않았으면 -1
// 정상적인 값이 반영되어 있으면 dp[sugarWeight] 출력
print(dp[sugarWeight] > 5000 ? -1 : dp[sugarWeight])

내가 처음으로 풀었던 dp문제인데, 당시에는 dp로 풀지 않아 새롭게 풀어봤다.

적절한 조합으로 최솟값을 찾아라, 최댓값을 찾아라 하는 문제가 주로 다이나믹 프로그래밍으로 보인다.

 

그리디 풀이[메모리: 69100KB, 시간: 8ms]

var sugarWeight = Int(readLine()!)!
var bag = 0

// 설탕이 5kg의 배수인지 확인.
// 5배수가 맞으면 종료
if sugarWeight % 5 == 0 {
    bag = sugarWeight / 5
    sugarWeight = 0
}
/// - IMPORTANT: 18kg은 3의 배수이지만, (3kg 1개, 5kg 3개)가 더 효율적이므로, 3의 배수는 확인하지 않음.

while sugarWeight > 0 {
    // 설탕 3kg씩 배낭에 담으면서
    sugarWeight -= 3
    bag += 1
    // 남은 설탕이 5의배수인지 확인.
    if sugarWeight % 5 == 0 {
        bag += sugarWeight / 5
        sugarWeight = 0
    }
}
// 3의 배수나 5의 배수의 합이 아니면 -1 출력
// 남은 설탕의 무게가 0이면 bag(몇봉지)인지 출력
print(sugarWeight < 0 ? -1 : bag)

5키로씩 더 효율적으로 담을 수 있는가? 라는게 이 문제의 핵심이라는 사실만 기억해도 쉽게 풀 수 있다.