[Java] BOJ 1931 회의실 배정

문제 

문제 링크 https://www.acmicpc.net/problem/1931

• 한 개의 회의실이 있는데 이를 사용하고자 하는 N개의 회의에 대하여 회의실 사용표를 만들려고 한다.
• 각 회의 I에 대해 시작시간과 끝나는 시간이 주어져 있고, 각 회의가 겹치지 않게 하면서 회의실을 사용할 수 있는 회의의 최대 개수를 찾아보자.
• 단, 회의는 한번 시작하면 중간에 중단될 수 없으며 한 회의가 끝나는 것과 동시에 다음 회의가 시작될 수 있다.
• 회의의 시작시간과 끝나는 시간이 같을 수도 있다. 이 경우에는 시작하자마자 끝나는 것으로 생각하면 된다.
• 첫째 줄에 회의의 수 N(1 ≤ N ≤ 100,000)이 주어진다.
• 둘째 줄부터 N+1 줄까지 각 회의의 정보가 주어지는데 이것은 공백을 사이에 두고 회의의 시작시간과 끝나는 시간이 주어진다. 시작 시간과 끝나는 시간은 231-1보다 작거나 같은 자연수 또는 0이다.
• 첫째 줄에 최대 사용할 수 있는 회의의 최대 개수를 출력한다.

아이디어

• 회의실 시간을 2차원 배열에 시작시간과 끝시간을 저장한다.
• 끝나는 시간이 빠를수록 많은 회의실을 배정할 수 있기 때문에 끝나는 시간을 기준으로 먼저 정렬한다.
• 그리고 끝시간이 같다면 시작시간을 기준으로 오름차순 정렬을 한다.
• 정렬을 다 하고 반복문을 돌리면서 현재 회의실 시작시간이 저장되어있던 끝난 시간보다 같거나 크다면
• 카운트를 해주고 끝나는 시간을 현재 회의실이 끝나는 시간으로 갱신시킨다.  

겪은 시행착오

• X (사실 풀면서 2차원 배열로 풀면 시간 초과나 메모리 초과가 나진 않을까 걱정을 좀 했다...)

맞아서 다행이다...ㅎ

코드

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

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

        // 회의실 개수
        int N = Integer.parseInt(br.readLine());

        // 회의실 시간 저장할 배열
        int office[][] = new int[N][2];

        for(int i = 0; i < N; i++){
            StringTokenizer st = new StringTokenizer(br.readLine());

            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());

            // 시작시간과 끝시간 배열에 저장
            office[i][0] = a;
            office[i][1] = b;
        }

        // 오름차순으로 정렬
        Arrays.sort(office, (n1, n2) -> {
            // 끝나는 시간을 기준으로 정렬 만약 비교하는 두 값이 같다면
            // 시작시간을 기준으로 정렬
            if((n1[1] - n2[1]) == 0){
                return n1[0] - n2[0];
            }

            return n1[1] - n2[1];
        });

        // 회의실 카운트
        int cnt = 0;
        // 끝나는 시간 계산할 때 쓸 변수
        int last_time = 0;

        for(int i = 0; i < N; i++){
            // 만약 해당 회의실 시작시간이 앞에 계산된 끝난 시간보다
            // 같거나 크다면
            if(office[i][0] >= last_time){
                // 회의실 + 1
                cnt = cnt + 1;
                // 끝난 시간을 해당 회의실 끝나는 시간으로 갱신
                last_time = office[i][1];
            }
        }

        System.out.println(cnt);
    }
}