Knowledge/Java

[백준] 1931번: 회의실 배정 JAVA 코드, Greedy

발달중인 망고 2023. 12. 1. 12:50

회의실 배정 성공

문제

한 개의 회의실이 있는데 이를 사용하고자 하는 N개의 회의에 대하여 회의실 사용표를 만들려고 한다. 각 회의 I에 대해 시작시간과 끝나는 시간이 주어져 있고, 각 회의가 겹치지 않게 하면서 회의실을 사용할 수 있는 회의의 최대 개수를 찾아보자. 단, 회의는 한번 시작하면 중간에 중단될 수 없으며 한 회의가 끝나는 것과 동시에 다음 회의가 시작될 수 있다. 회의의 시작시간과 끝나는 시간이 같을 수도 있다. 이 경우에는 시작하자마자 끝나는 것으로 생각하면 된다.

입력

첫째 줄에 회의의 수 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N+1 줄까지 각 회의의 정보가 주어지는데 이것은 공백을 사이에 두고 회의의 시작시간과 끝나는 시간이 주어진다. 시작 시간과 끝나는 시간은 231-1보다 작거나 같은 자연수 또는 0이다.

출력

첫째 줄에 최대 사용할 수 있는 회의의 최대 개수를 출력한다.

예제 입력

11
1 4
3 5
0 6
5 7
3 8
5 9
6 10
8 11
8 12
2 13
12 14

예제 출력

4

시간이 엄청 들어서 풀었네요 접근 방법을 잘못 골라서 시간을 많이 잡아먹었는데 문제를 많이보고 유형을 알아보는 게 중요한 것 같습니다ㅠㅠ

먼저 틀려서 애먹었던 접근 방법을 설명드리겠습니다.

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

public class P_1931 {
    public static void main(String[] args) throws IOException {
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        int row = Integer.parseInt(bf.readLine());
        int[][] time = new int[row][2];
        int maxNum = 0;
        int answer = 0;

        for (int i = 0; i < row; i++) {
            String[] inputTime = bf.readLine().split(" ");
            for (int j = 0; j < 2; j++) {
                time[i][j] = Integer.parseInt(inputTime[j]);
                if (time[i][j] > maxNum) {
                    maxNum = time[i][j];
                }
            }
        }

        boolean[] timeCheck = new boolean[maxNum+1];

        Arrays.sort(time, Comparator.comparingInt(a -> a[1] - a[0]));

        for (int i = 0; i < row; i++) {
            if (canInsert(timeCheck, time, i)) {
                answer ++;
            }
        }
        System.out.println(answer);
    }

    public static boolean canInsert(boolean[] timeCheck, int[][] time, int index) {
        for (int i = time[index][0]; i < time[index][1]; i++) {
            if (timeCheck[i]) {
                return false;
            }
        }
        for (int i = time[index][0]; i < time[index][1]; i++) {
            timeCheck[i] = true;
        }
        return true;
    }
}

저는 Greedy 문제이기 때문에 가장 회의시간을 적게 잡아먹는것으로 정렬하고 적은 시간을 탐색하면서 회의실의 시간표를 채워가는 형식으로 진행했습니다. 물론 답을 도출해내기는 하지만 큰 문제가 있습니다.

바로 시작하자마자 끝나는 회의를 회의시간에 배정하지 못하기 때문에 문제를 풀 수 없었습니다.

5
1 5
0 3
2 2
2 2
2 2

원래 답은 3이 나와야 하는 것이 맞지만 2 2라는 input 때문에 2시의 회의시간을 꽉 채운다면 다른 2 2들이 들어가지 못합니다. 하지만 그렇다고 회의시간을 처리하지 않는다면 0시에서 3시까지의 회의도 통과되는 오류가 있습니다.

그래서 약간 힌트를 봤는데 힌트는 회의가 끝나는 순서로 정렬해서 푸는 게 가장 효과적이다 라는 것이었습니다.

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

public class P_1931 {
    public static void main(String[] args) throws IOException {
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        int row = Integer.parseInt(bf.readLine());
        int[][] time = new int[row][2];

        for (int i = 0; i < row; i++) {
            String[] input = bf.readLine().split(" ");
            time[i][0] = Integer.parseInt(input[0]);
            time[i][1] = Integer.parseInt(input[1]);
        }

        // 종료 시간을 기준으로 정렬, 종료 시간이 같으면 시작 시간을 기준으로 정렬
        Arrays.sort(time, (a, b) -> {
            if (a[1] == b[1]) {
                return a[0] - b[0];
            }
            return a[1] - b[1];
        });

        int answer = 0;
        int end = -1;
        for (int[] meeting : time) {
            if (meeting[0] >= end) {
                end = meeting[1];
                answer++;
            }
        }

        System.out.println(answer);
    }
}

알고리즘 유형을 많이 보는 게 중요한 것 같습니다.