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

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)

출력

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

---

그리디 알고리즘입니다. 큐를 이용해 큐 자료구조 한 수업의 끝시간이 들어가있다면 강의실을 사용중인 것으로 표현했습니다. 예를들어 큐에 두 개의 값이 들어가있다면 두 개의 강의실이 사용중이라는 의미입니다. 

앞에 수업이 끝날때마다 큐에서 앞의 수업을 빼고 뒤에 수업을 넣어주는 방식을 택했습니다.

 

먼저 시작시간을 기준으로 강의수업을 오름차순으로 정렬을 해줍니다.

 for(int i=0;i<n;i++)
    {
        int a,b;
        cin>>a>>b;
        v.push_back(make_pair(a,b));
    }
    
    sort(v.begin(),v.end());

하지만 시작시간과 끝나는시간이 강의마다 상이하기 때문에 우선순위 큐를 이용해 작은 값부터 즉 먼저 끝나는 수업이 나오게 하여 다음 강의와 비교하는 방식으로 구현하였습니다.

 

전체 소스코드

#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;

int main()
{
    int n;
    
    vector<pair<int,int>>v;
    priority_queue<int,vector<int>,greater<int>>pq;
    
    cin>>n;
    
    for(int i=0;i<n;i++)
    {
        int a,b;
        cin>>a>>b;
        v.push_back(make_pair(a,b));
    }
    
    sort(v.begin(),v.end());
    
    pq.push(v[0].second);
    int answer = 1;
    for(int i=1;i<n;i++)
    {
        if(pq.top()<=v[i].first)
        {
            pq.pop();
            pq.push(v[i].second);
        }
        else
        {
            pq.push(v[i].second);
        }
    }
    cout<<pq.size()<<endl;
    
    return 0; 
}