[Java] 백준 2839번. 설탕 배달

[Java] 백준 2839번. 설탕 배달

 

문   제 : 

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

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

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

 

입   력 : 

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

 

출   력 : 

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

 

 

Example )

Input: 18
Output: 4
///////////////////////////////////////////////
Input: 4
Output: -1
///////////////////////////////////////////////
Input: 6
Output: 2
///////////////////////////////////////////////
Input: 9
Output: 3
///////////////////////////////////////////////
Input: 11
Output: 3

 

---

접   근 : 

DP를 활용하여

배달할 설탕을 1씩 늘려가면서 최소 설탕봉지의 수를 업데이트 해준다.

---

해   결 : 

설탕이 n kg 일 때, 최소로 필요한 설탕봉지를 저장하는 배열 dp[n]을 선언하고 매우 큰 값으로 초기화 한다.

 

반복문 안에서 배달해야 하는 설탕을 1kg씩 늘려준다.

설탕 봉지 두 종류 - 5kg, 3kg을 둘 다 사용하면서 비교한다.

만약 배달할 수 있는 설탕 봉지가 있다면 그 봉지로 배달한다고 가정하고 그 kg만큼을 뺀 나머지 배달의 최소값과 더해준다.

 

---

dp 배열 선언 & 초기화

 

int dp[] = new int[n+1];
Arrays.fill(dp, Integer.MAX_VALUE-1);
dp[0] = 0;

 

 

---

 

설탕 봉지의 종류 담을 배열선언, 초기화

 

 int bag[] = new int[]{5, 3};

 

---

반복문을 돌면서 업데이트 해준다.

 

 

for(int i=1; i<n+1; i++){
    for(int j=0; j<2; j++){
        if(i >= bag[j] && dp[i-bag[j]] < dp[i] ){
            dp[i] = dp[i-bag[j]] + 1;
        }
    }
}

 

구하려는 값은 dp[n]이다. 

만약 dp[n]을 운반할 수 없다면 처음 초기화 했던 값 그대로 일것이다. 

---

결   과  :

import java.util.Scanner;
import java.util.*;
public class Main {

    public static void main(String[] args) {
        int n;
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();

        int dp[] = new int[n+1];
        Arrays.fill(dp, Integer.MAX_VALUE-1);
        dp[0] = 0;

        int bag[] = new int[]{5, 3};

        for(int i=1; i<n+1; i++){
            for(int j=0; j<2; j++){
                if(i >= bag[j] && dp[i-bag[j]] < dp[i] ){
                    dp[i] = dp[i-bag[j]] + 1;
                }
            }
        }
        if(dp[n] == Integer.MAX_VALUE-1) System.out.println(-1);
        else System.out.println(dp[n]);
    }

}