[백준 - G5] 1931. 회의실 배정

문제 링크

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

문제 설명

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

입력

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

출력

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

 

 

 

---

아이디어

회의가 빨리 끝나는 순으로 정렬하고 같다면, 회의가 빨리 시작하는 순으로 정렬한다.

• times[i][1] (end) 먼저 오름차순으로 정렬 후, times[i][0] (start) 로 정렬한다.

제출 코드

class Solution {

    public static void main(String[] args) throws Exception {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());

        int[][] times = new int[N][2];

        for (int i = 0; i < N; i++) {
            String[] split = br.readLine().split(" ");

            int start = Integer.parseInt(split[0]);
            int end = Integer.parseInt(split[1]);

            times[i][0] = start;
            times[i][1] = end;

        }

        // 끝나는 시간이 빠른 순으로 정렬해야
        Arrays.sort(times, new Comparator<int[]>() {

            @Override
            public int compare(int[] o1, int[] o2) {
                int cmp = Integer.compare(o1[1], o2[1]); // 오름차순

                if (cmp == 0) {
                    return Integer.compare(o1[0], o2[0]);
                }

                return cmp;
            };
        });

        // System.out.println(Arrays.deepToString(times));

        int cnt = 1;
        int last = times[0][1];
        for (int i = 0; i < N - 1; i++) {
            if (last <= times[i + 1][0]) {
                // System.out.println(times[i + 1][0] + " " + times[i + 1][1]);
                last = times[i + 1][1];
                cnt++;
            }
        }

        System.out.println(cnt);
    }

}

정렬 부분 Stream으로 구현

• Comparator.comparing : 정렬 기준 하나 만들 때 사용
• (int[] arr) -> arr[1] : 배열의 두 번째 원소를 꺼내는 함수
• .thenComparing(arr -> arr[0]) : 끝나는 시간이 같으면, 시작 시간(arr[0]) 기준 오름차순

Arrays.sort(times, Comparator
    .comparing((int[] arr) -> arr[1])
    .thenComparing(arr -> arr[0])
);