[Baekjoon/C++] 11000번 - 강의실 배정

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

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

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

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

 

입력

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

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

 

출력

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

 

---

풀이

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

struct Class {
	int start;
	int time;
};

int N;
Class arr[200001];
priority_queue<int> q;

bool compare(Class a, Class b);

bool compare(Class a, Class b) {
	if (a.start == b.start) return a.time < b.time;

	return a.start < b.start;
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);

	cin >> N;
	for (int i = 0; i < N; i++) 
		cin >> arr[i].start >> arr[i].time;

	sort(arr, arr + N, compare);
		
	for (int i = 0; i < N; i++) {
		q.push(-arr[i].time);

		if (-q.top() <= arr[i].start) q.pop();
	}

	cout << q.size() << '\n';

	return 0;
}

 

bool compare(Class a, Class b) {
	if (a.start == b.start) return a.time < b.time;

	return a.start < b.start;
}

 오름차순으로 수업을 정렬하는 함수이다.

 

	for (int i = 0; i < N; i++) {
		q.push(-arr[i].time);

		if (-q.top() <= arr[i].start) q.pop();
	}

	cout << q.size() << '\n';

 우선순위 큐를 사용하여 필요한 강의실의 수를 구하는 코드이다. 우선순위 큐는 내림차순으로 정렬이 되고, 해당 문제에서는 작은 것부터 꺼내야 하기에 수에 -를 붙여서 push한다.

 큐의 top(가장 빨리 끝나는 수업)보다 이번에 탐색하는 수업 i의 시작 시간이 크거나 같으면 top 수업이 끝난 뒤에 i 수업을 할 수 있는 것이므로 q의 top을 제거한다.

 for문이 끝나고 q에 남아 있는 수의 개수가 필요한 강의실의 개수이다.