[Algorithm] BAEKJOON 11000번: 강의실 배정 (C++)

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

 

 

 

 

문제

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

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

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

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

 

입력

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

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

 

출력

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

 

 

 

 

 

풀이

그리디 알고리즘과 우선순위 큐를 활용해서 가장 빨리 끝나는 수업을 효율적으로 추적하여 해결하였다.

 

 

C++ 소스 코드

#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>

using namespace std;

// 수업 정보를 저장하는 구조체
struct Class {
    int start, end;
};

// 시작 시간 기준으로 정렬하기 위한 비교 함수
bool compare(const Class& a, const Class& b) {
    return a.start < b.start;
}

int main() {
    int N;
    cin >> N;

    vector<Class> classes(N);

    for (int i = 0; i < N; i++) {
        cin >> classes[i].start >> classes[i].end;
    }

    // 수업을 시작 시간 기준으로 정렬
    sort(classes.begin(), classes.end(), compare);

    // 최소 힙 (priority queue) 사용, 끝나는 시간을 기준으로 관리
    priority_queue<int, vector<int>, greater<int> > pq;

    // 첫 번째 수업의 끝나는 시간을 추가
    pq.push(classes[0].end);

    for (int i = 1; i < N; i++) {
        // 현재 수업의 시작 시간이 가장 빨리 끝나는 수업 이후라면, 같은 강의실 사용 가능
        if (pq.top() <= classes[i].start) {
            pq.pop();
        }
        // 현재 수업의 끝나는 시간을 큐에 추가
        pq.push(classes[i].end);
    }

    // 큐의 크기가 필요한 강의실의 수
    cout << pq.size() << endl;

    return 0;
}