[Swea] D3 1860 : 진기의 최고급 붕어빵 (java)

🧪 1860 진기의 최고급 붕어빵

---

난이도 : D3
유형 : 구현

SW Expert Academy

 

📝 문제

---

진기는 붕어빵 가게를 운영하고 있다.

진기가 파는 붕어빵은 그냥 붕어빵이 아니라 겉은 바삭! 속은 말랑! 한입 물면 팥 앙금이 주르륵 흘러 입안에서 춤을 추며,

절로 어릴 적 호호 불며 먹었던 뜨거운 붕어빵의 추억이 떠올라 눈물이 나오게 되는 최고급 붕어빵이다.

진기는 이런 붕어빵을 보통 사람들에게는 팔지 않는다.

그는 무조건 예약제로만 손님을 받으며, 예약을 하려는 손님들은 진기의 까다로운 자격 검증에서 합격해야만 붕어빵을 맛 볼 자격을 얻는다.

그래서 오늘은 N명의 사람이 자격을 얻었다.

진기는 0초부터 붕어빵을 만들기 시작하며, M초의 시간을 들이면 K개의 붕어빵을 만들 수 있다.

서빙은 진기가 하는 것이 아니기 때문에, 붕어빵이 완성되면 어떤 시간 지연도 없이 다음 붕어빵 만들기를 시작할 수 있다.

0초 이후에 손님들이 언제 도착하는지 주어지면, 모든 손님들에게 기다리는 시간없이 붕어빵을 제공할 수 있는지 판별하는 프로그램을 작성하라.

 

입력

 

첫 번째 줄에 테스트 케이스의 수 T가 주어진다.

각 테스트 케이스의 첫 번째 줄에는 세 자연수 N, M, K(1 ≤ N, M, K ≤ 100)가 공백으로 구분되어 주어진다.

두 번째 줄에는 N개의 정수가 공백으로 구분되어 주어지며,

각 정수는 각 사람이 언제 도착하는지를 초 단위로 나타낸다. 각 수는 0이상 11,111이하이다.

 

출력

 

각 테스트 케이스마다 ‘#x’(x는 테스트케이스 번호를 의미하며 1부터 시작한다)를 출력하고,

모든 손님에 대해 기다리는 시간이 없이 붕어빵을 제공할 수 있으면 “Possible”을, 아니라면 “Impossible”을 출력한다.

 

🚧  주의할 점

---

1. 도착시간이 중복되는 경우

ex) 2 2 둘 다 2초에 왔을 때를 고려해서 문제를 풀이해야 한다.

 

중복을 고려하지 못해 fail이 발생한 경우 ( 1000개 中 987개 )

import java.io.*;
import java.util.*;

public class Solution {

    static int[] arriveTime, dp;
    public static void main(String[] args) throws Exception {

//        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedReader br = new BufferedReader(new FileReader("input.txt"));

        int t = Integer.parseInt(br.readLine());

        for (int i = 1; i <= t; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int n = Integer.parseInt(st.nextToken());
            int m = Integer.parseInt(st.nextToken());
            int k = Integer.parseInt(st.nextToken());
            int num = 1;
            int res = 1;
            arriveTime = new int[n];

            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < n; j++) {
                arriveTime[j] = Integer.parseInt(st.nextToken());
            }

            timeSort();
            dp = new int[arriveTime[n - 1] + 1];

            for (int idx = 1; idx <= arriveTime[n - 1]; idx++) {
                dp[idx] = dp[idx - 1];

                if (idx % m == 0) dp[idx] += k;
            }

            StringBuilder sb = new StringBuilder("#" + i);
            for (int value : arriveTime) {
                dp[value] -= num++; // 이 부분에서 문제가 발생한다.

                if (dp[value] < 0) {
                    res = 0;
                    break;
                }
            }

            if (res == 0) {
                sb.append(" ").append("Impossible");
            } else {
                sb.append(" ").append("Possible");
            }

            System.out.println(sb);
        }

        br.close();
    }

    private static void timeSort() {

        for(int index = 1; index < arriveTime.length ; index++){
            int temp = arriveTime[index];
            int prev = index - 1;
            while( (prev >= 0) && (arriveTime[prev] > temp) ) {
                arriveTime[prev+1] = arriveTime[prev];
                prev--;
            }

            arriveTime[prev + 1] = temp;
        }
    }
}

 

🧐 핵심 로직

---

◈  도착시간에 대해서 오름차순으로 정렬해준다. (Arrays.sort로 가능하지만 삽입정렬을 활용했다.)

◈  손님이 없다고 가정하고, m초 마다 k개의 붕어빵을 만들었을 때 각 초마다의 붕어빵 개수를 dp배열안에 저장한다.

◈  오름차순으로 정렬된 도착시간을 반복하며 앞의 사람 수 마다 num값을 증가시켜 방문한 시간의 ( 붕어빵 개수 - 앞의 사람 수 {num} ) 이 0보다 작은 경우 Impossible을 아니라면 Possible을 출력하도록 한다.

💻 최종 코드 (254 ms)

---

import java.io.*;
import java.util.*;

public class Solution {

    static int[] arriveTime, dp;
    public static void main(String[] args) throws Exception {

//        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedReader br = new BufferedReader(new FileReader("input.txt"));

        int t = Integer.parseInt(br.readLine());

        for (int i = 1; i <= t; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int n = Integer.parseInt(st.nextToken());
            int m = Integer.parseInt(st.nextToken());
            int k = Integer.parseInt(st.nextToken());
            int num = 0;
            int res = 1;
            arriveTime = new int[n];

            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < n; j++) {
                arriveTime[j] = Integer.parseInt(st.nextToken());
            }

            timeSort();
            dp = new int[arriveTime[n - 1] + 1];

            for (int idx = 1; idx <= arriveTime[n - 1]; idx++) {
                dp[idx] = dp[idx - 1];

                if (idx % m == 0) dp[idx] += k;
            }

            StringBuilder sb = new StringBuilder("#" + i);
            for (int value : arriveTime) {
                num++; // 문제 발생 부분 수정

                if (dp[value] - num < 0) {
                    res = 0;
                    break;
                }
            }

            if (res == 0) {
                sb.append(" ").append("Impossible");
            } else {
                sb.append(" ").append("Possible");
            }

            System.out.println(sb);
        }

        br.close();
    }

    private static void timeSort() {

        for(int index = 1; index < arriveTime.length ; index++){
            int temp = arriveTime[index];
            int prev = index - 1;
            while( (prev >= 0) && (arriveTime[prev] > temp) ) {
                arriveTime[prev+1] = arriveTime[prev];
                prev--;
            }

            arriveTime[prev + 1] = temp;
        }
    }
}