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

문제 링크

2839번: 설탕 배달

문제

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

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

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

입력

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

출력

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

문제 풀이

상근이가 설탕을 최소한의 이동으로 옮기고 싶다면 5킬로그램 봉지를 최대한 많이 가져가면 된다.

하지만 설탕의 무게를 정확하게 배달해야된다는 조건이 있다.

예를 들어 14킬로그램의 설탕을 옮기고 싶다면 5킬로그램 3개를 옮기면 안된다.

즉 5킬로그램 1개와 3킬로그램 2개를 옮겨야 하기 때문에 모든 무게에 대하여 5킬로그램부터 고려하면 안된다.

 

어떠한 무게에 대해 정확하게 옮길 수 있는지 알아보는게 먼저일 것 같다.

이에 대하여 규칙을 찾던 도중 어떠한 숫자에서 3을 계속 빼다가 그 숫자가 5의 배수가 되면 그때가 정확하게 최소로 옮길 수 있는 지점인 것을 발견했다.

 

예를 들어 16kg를 옮긴다고 해보자.

16 - 3 = 13

13 - 3 = 10

이 때 10은 5의 배수이니까 이 때가 정확하게 최소로 옮길 수 있는 지점이다.

 

여기서 계속 빼다가 음수가 된다면  이는 옮길 수 없는 지점이기 때문에 -1을 반환하면 된다.

이 규칙을 사용해서 문제를 해결했다.

소스 코드

func solution() -> Int{
    var n = Int(readLine()!)!
    var result = 0
    
    if n % 5 == 0{
        return n / 5
    }
    
    while (n >= 0){
        n -= 3
        result += 1
        if n % 5 == 0{
            return result + (n / 5)
        }
    }
    
    
    return -1
}

print(solution())