코딩은 마라톤

[백준] 2839번 : 설탕 배달 – JAVA [자바] 본문

CS/알고리즘

[백준] 2839번 : 설탕 배달 – JAVA [자바]

anxi 2023. 10. 10. 15:33

[Silver IV] 설탕 배달 - 2839

문제 링크

성능 요약

메모리: 18804 KB, 시간: 236 ms

분류

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

문제 설명

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

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

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

입력

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

출력

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

 


  • 문제 풀이
그리디 알고리즘 문제이지만, 수학 방정식을 이용해 문제를 풀었다.
3x + 5y = N
  1. N을 입력받는다.
  2. N을 3과 5로 각각 나눈 몫을 range변수에 넣는다. (배열의 범위를 지정하는데 배열의 크기가 range값을 초과하지 않기 때문이다.)
  3. 위의 방정식에서 x에 해당하는 list3과 y에 해당하는 list5를 만든 후 현재 index의 값이 배열의 값이 되게 만들어준다. 
  4. 3x + 5y에 해당하는 x, y값을 구하기 위해 반복문을 2번 돌려 x, y의 합을 sum에 넣어준다. 
  5. 만약 sum이 비어있지 않다면 sum을 오름차순으로 정렬한 다음, 합이 최소인 값을 출력한다.
  6. 비어있다면 -1을 출력한다.

  • 코드
import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();

        int range3 = n / 3;
        int range5 = n / 5;

        int[] list3 = new int[range3 + 1];
        int[] list5 = new int[range5 + 1];

        for(int i = 0; i <= range3; i++){
            list3[i] = i;
        }
        for(int j = 0; j <= range5; j++){
            list5[j] = j;
        }

        ArrayList<Integer> sum = new ArrayList<>();

        for(int i = 0; i <= range3; i++){
            for(int j = 0; j <= range5; j++){
                if((3 * list3[i]) + (5 * list5[j]) == n){
                    sum.add(i+j);
                }
            }
        }
        
        if(!sum.isEmpty()){
            Collections.sort(sum);
            System.out.println(sum.get(0));
        }else System.out.println(-1);
    }
}