[백준] 11000번 강의실 배정 - C++

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

11000번: 강의실 배정

문제

수강신청의 마스터 김종혜 선생님에게 새로운 과제가 주어졌다. 

김종혜 선생님한테는 Si에 시작해서 Ti에 끝나는 N개의 수업이 주어지는데, 최소의 강의실을 사용해서 모든 수업을 가능하게 해야 한다. 

참고로, 수업이 끝난 직후에 다음 수업을 시작할 수 있다. (즉, Ti ≤ Sj 일 경우 i 수업과 j 수업은 같이 들을 수 있다.)

수강신청 대충한 게 찔리면, 선생님을 도와드리자!

입력

첫 번째 줄에 N이 주어진다. (1 ≤ N ≤ 200,000)

이후 N개의 줄에 Si, Ti가 주어진다. (0 ≤ Si < Ti ≤ 109)

출력

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

---

시간 배분에 관한 문제를 보면 그리디 알고리즘이 떠오른다.

  입력된 인풋을 시작 순서대로 정렬을 한 다음 첫 번째 원소의 종료 시간을 우선순위 큐에 넣는다. 이렇게 되면 우선순위 큐의 top 원소는 현재 강의실 중 가장 빨리 끝나는 강의실의 시간이 될 것이다.

  이때 top 보다 현재 강의의 시작시간이 이르다면 새로운 강의실이 필요하다. 따라서 강의실의 개수를 하나 늘리고 이 강의의 끝나는 시간을 우선순위 큐에 넣어준다.

  반대로 top 보다 현재 강의의 시작시간이 늦거나 같다면 해당 강의실에서 강의를 진행할 수 있다. 따라서 강의실의 개수를 늘릴 필요 없이 top을 pop 하고 해당 강의실의 새로운 종료시간이 현재 강의의 종료시간을 push 한다.

#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
typedef pair<int,int> point;
int N;
vector <point> arr;
priority_queue <int,vector<int>,greater<int> > q;
int a,b;
int ans = 1;

int main(){
    cin >> N;
    for(int i = 0 ; i < N ; i++){
        cin >> a >> b;
        arr.push_back(make_pair(a,b));
    }
    sort(arr.begin(),arr.end());
    q.push(arr[0].second);
    for(int i = 1; i < N ; i++){
        int top = q.top();
        int curs = arr[i].first;
        int curt = arr[i].second;
        if(curs < top){
            ans++;
            q.push(curt);
        }
        else{
            q.pop();
            q.push(curt);
        }
    }
    cout << ans;
}

---

문제를 보자마자 그리디 문제인 것을 알았지만 정렬을 한 다음 어떻게 풀지는 잠시 고민을 했어야 했다. 하지만 이내 큐로 풀 수 있을 것 같았다. 그래서 큐를 활용하여 코드를 작성하였고 예제 문제에 대한 정답은 잘 나왔다. 하지만 틀렸습니다가 나와 조금 더 생각을 해본 결과 처음에 생각을 할 때에는 정렬을 하였으니 굳이 PQ가 아니어도 front가 가장 작은 수라고 생각했지만 시작 시간을 기준으로 정렬을 하였고 큐에는 종료시간을 기준으로 push 하므로 PQ를 활용하여야 될 것 같다는 생각이 들어 PQ로 정정하였고 정답을 받아 낼 수 있었다. 이번 문제 역시 생각을 적어가면서 문제를 풀었더니 코드가 잘 짜인 것 같다. 쉬운 문제를 푼 덕도 있겠지만 확실히 코드가 깔끔하게 짜이는 것 같다!