[Baekjoon 2839] 설탕 배달 - Java

[Silver IV] 설탕 배달 - 2839

문제 링크

성능 요약

메모리: 14212 KB, 시간: 120 ms

분류

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

제출 일자

2024년 3월 7일 10:16:53

문제 설명

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

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

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

입력

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

출력

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

문제 풀이

그리디 문제이기 때문에 점화식을 사용해도 된다.

필자의 알고리즘은 이러하다.

5kg과 3kg 중 우선 순위는 5kg이다.

입력받은 N이 5로 나눠 떨어지면 그대로 개수를 출력해주는게 최소값을 가진다.

입력받은 N이 3으로 나눠 떨어지면 N에서 3만큼 빼준다.

만약 3과 5로 나눠 떨어지지 않는다면 3을 빼준다.

위 과정을 반복하면 된다.

만약 3과 5로 나눠 떨어지지 않을 경우 N은 0이 되지 않는다.

삼항 연산자를 사용해 N이 0인지 확인하고 0이 아니라면 -1을 출력하면 된다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int N = Integer.parseInt(br.readLine());
        int cnt = 0;

        while (N >= 3) {
            if (N % 5 == 0) {
                cnt += (N / 5);
                N = 0;
            }
            else if (N % 3 == 0) {
                cnt ++;
                N -= 3;
            }
            else {
                N -= 3;
                cnt ++;
            }
        }

        System.out.println(N == 0 ? cnt : -1);
    }
}

코드 풀이

int N = Integer.parseInt(br.readLine());
int cnt = 0;

• N: 입력받는 설탕을 배달해야하는 무게
• cnt: 배달해야하는 횟수

while (N >= 3) {
    if (N % 5 == 0) {
        cnt += (N / 5);
        N = 0;
    }
    else if (N % 3 == 0) {
        cnt ++;
        N -= 3;
    }
    else {
        N -= 3;
        cnt ++;
    }
}

• if를 사용해서 N이 3, 5로 나눠 떨어지는지 확인하고 나눠 떨어지면 그에 상응하여 cnt를 더해줬다.
• 나눠 떨어지지 않는다면 3을 빼준다.