[백준][11000번: 강의실 배정]

문제

문제링크

수강신청의 마스터 김종혜 선생님에게 새로운 과제가 주어졌다.
김종혜 선생님한테는 Si에 시작해서 Ti에 끝나는 N개의 수업이 주어지는데, 최소의 강의실을 사용해서 모든 수업을 가능하게 해야 한다.
참고로, 수업이 끝난 직후에 다음 수업을 시작할 수 있다. (즉, Ti ≤ Sj 일 경우 i 수업과 j 수업은 같이 들을 수 있다.)
수강신청 대충한 게 찔리면, 선생님을 도와드리자!

입력

첫 번째 줄에 N이 주어진다. (1 ≤ N ≤ 200,000)
이후 N개의 줄에 Si, Ti가 주어진다. (0 ≤ Si < Ti ≤ 109)

출력

강의실의 개수를 출력하라.

접근 방법

처음에 arrayList에 끝나는 시간을 담고 우선순위 큐에 시작 시간을 넣었다.
비교를 하면서 remove와 add를 하면서 구했다. 하지만 시간초과가 나서 고민을 해봤다.
문제는 remove라는게 시간을 많이 잡기 때문에 시간초과가 났던 것이였다. 그래서 반대로 해봤다.
배열에 넣고 시작시간을 기준으로 오름차순 정렬을 한뒤 우선순위 큐에 끝나는 시간을 넣어서 비교를 하면서 했더니 통과하였다.

코드

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

public class Main {
    static int N;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        Lecture[] lectures = new Lecture[N];
        for(int i=0;i<N; i++){
            StringTokenizer st = new StringTokenizer(br.readLine());
            int S = Integer.parseInt(st.nextToken());
            int T = Integer.parseInt(st.nextToken());
            lectures[i] = new Lecture(S,T);
        }
        Arrays.sort(lectures, (o1,o2)->Integer.compare(o1.start,o2.start)); // 시작 기준으로 정렬
        PriorityQueue<Integer> pq = new PriorityQueue<>(); // 우선순위 큐
        pq.add(lectures[0].end); // 처음 시작 강의의 끝나는 시간
        for(int i=1; i<N; i++){
            int end = pq.peek();
            if(end <= lectures[i].start){ // 진행중인 강의의 끝나는 시간이 다음 강의 시작 시간보다 작을 때
                pq.poll();
            }
            pq.add(lectures[i].end); // 다음 강의의 끝나는 시간 추가
        }
        System.out.println(pq.size()); // 우선순위 큐에 size가 강의실를 최소한 사용할 수 있는 개수
    }
    static class Lecture{
        int start;
        int end;
        public Lecture(int start, int end){
            this.start = start;
            this.end = end;
        }
    }
}