[Java] 백준 1931 - 회의실 배정

Baekjoon_1931 - 회의실 배정

https://www.acmicpc.net/problem/1931

문제

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

입력

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

출력

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

---

해결방법

• 대표적인 그리디 알고리즘 문제로, 일찍 끝나는 회의를 선택한 후 그 회의와 겹치는 다를 회의를 제외하는 과정을 반복하면 된다.
  • 예를 들어, 회의 네개가 (1, 3), (2, 4), (5, 7), (7, 10) 으로 정해져 있다고 가정하자.
    • 가장 빨리 끝나는 회의인 (1, 3) 선택 후, 겹지는 회의인 (2, 4) 제외
    • 그다음으로 빨리 끝나는 회의인 (5, 7) 선택 (겹치는 회의 없음)
    • 그다음으로 빨리 끝나는 회의인 (7, 10) 선택
    • 따라서 가능한 회의의 최대 개수는 3개
• [시작시간, 종료시간] 을 각 원소로 가지는 arr 배열을 정렬하기 위해 Comparator 인터페이스를 구현한다.
  • 종료시간을 기준으로 정렬하되, 종료시간이 같으면 시작시간으로 정렬한다.
  • 만약 회의 세개가 (5, 5), (1, 3), (3, 5) 순서로 입력된다고 하자.
    • 종료시간을 기준으로 정렬하면 (1, 3), (5, 5), (3, 5) 가 되어버리는데 이렇게 되면, 가능한 회의 개수는 3개인데 결과는 2가 나올 것이다.
    • 따라서 종료시간이 같으면 시작시간으로 한번 더 정렬해서 (1, 3), (3, 5), (5, 5) 가 되야한다.

코드

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

public class Baekjoon_1931 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        int[][] arr = new int[n][2];

        StringTokenizer st;

        for (int i = 0; i < n; i++) {
            st = new StringTokenizer(br.readLine(), " ");
            arr[i][0] = Integer.parseInt(st.nextToken());
            arr[i][1] = Integer.parseInt(st.nextToken());
        }

        Arrays.sort(arr, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                if(o1[1] == o2[1]){
                    return o1[0] - o2[0];
                }
                return o1[1] - o2[1];
            }
        });

        int cnt = 0;
        int prev_end_time = 0;

        for (int i = 0; i < n; i++) {
            if (prev_end_time <= arr[i][0]) {
                prev_end_time = arr[i][1];
                cnt++;
            }
        }

        System.out.println(cnt);
    }
}