알고리즘(C++) / 백준 11000 : 강의실 배정

11000

www.acmicpc.net/problem/11000

11000번: 강의실 배정

 

문제

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

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

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

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

 

입력

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

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

 

출력

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

 

코드

//11000 강의실 배정
#include <iostream>
#include <algorithm>
#include <vector>
#include <utility> //pair 포함 헤더
#include <queue>
using namespace std;

int main() {
	int N;
	vector<pair <int, int>> schedule;
	cin >> N;

	int S, T;
	pair<int, int> p;
	for (int i = 0; i < N; i++) {
		cin >> S >> T;
		p.first = S;
		p.second = T;
		schedule.push_back(p);
	}

	sort(schedule.begin(), schedule.end()); //정렬

	priority_queue<int, vector<int>, greater<int>> pq; //오름차순으로 정렬되는 큐

	pq.push(schedule[0].second);

	for (int i = 1; i < N; i++) {
		if (pq.top() <= schedule[i].first)
			pq.pop();
		pq.push(schedule[i].second);
	}

	cout << pq.size() << endl;

	return 0;
}

 

설명

vector<pair<int, int>> 변수인 schedule을 입력받는다. first에는 시작 시간을 입력받고, seconde에는 끝나는 시간을 입력받는다. 

sort함수를 사용하여 시작 시간을 순서대로 오름차순으로 정렬해준다.

강의실 수를 나타내기 위해 priority_queue(우선순위 큐)를 사용한다. pq에는 강의가 끝나는 시간이 오름차순으로 정렬되어있다. 

 

처음 pq에 schedule 처음 배열의 끝나는 시간을 저장한다. pq에는 항상 작은 수부터 순서대로 저장되어있으므로 top과 schedule[i].first(시작 시간)과 비교하여 top보다 크다면 강의실을 이어받아 수업을 진행할수 있고, 그렇지 않다면 강의실이 하나 더 필요하다.

 

예를 들어 설명한다. 

5

1 3

2 4

3 5

4 7

8 10

pq에 3이 먼저 저장된다.

	start	i = 1	i = 2	i = 3	i = 4
pq	3	3	4	5	7
		4	5	7	10

pq에 i에 따라 다음과 같이 저장될 것이다.

push로 인해 바뀌지만 바로 오름차순으로 정렬된다.

 

---

 

오름차순으로 정렬되는 우선순위 큐이다.

priority_queue<자료형, container, 비교함수> 변수명;

priority_queue<int, vector<int>, greater<int>> pq; //오름차순

내림차순으로 정렬은 다음과 같이 한다.

priority_queue<자료형> 변수명;

priority_queue<int> pq; //내림차순

다음과 같이 우선순위 큐(priority_queue)를 선언할 수 있다. 

 

출력

 

고찰

이번 문제에서 pair와 priority_queue를 처음 사용하여 구현해보았다. 

처음에 문제를 풀었을 때는 우선순위 큐를 사용하지 않고 vector와 pair만을 사용하여 다음과 같이 구현하였다.

	/*
	int room_num = 0;
	int end = 0;
	while (schedule.size() !=0) {
		int i = 0;
		end = 0;
		while (i < schedule.size()) {
			if (end <= schedule[i].first) {
				end = schedule[i].second;
				schedule.erase(schedule.begin() + i); //해당하는 위치의 원소를 지운다.
			}
			else
				i++;
		}
		room_num++;
	}

	cout << room_num << endl;
	*/

하지만 다음과 같이 구현했을 때는 이중 반복문을 돌기 때문에 시간초과라는 오류가 났다.

그래서 이중 반복문을 사용하지 않도록 우선순위 큐를 사용하였다. 

STL 사용에 더 익숙해져야할 것 같다.