[1931] 회의실 배정

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

1931번: 회의실 배정

문제

한 개의 회의실이 있는데 이를 사용하고자 하는 N개의 회의에 대하여 회의실 사용표를 만들려고 한다. 각 회의 I에 대해 시작시간과 끝나는 시간이 주어져 있고, 각 회의가 겹치지 않게 하면서 회의실을 사용할 수 있는 회의의 최대 개수를 찾아보자. 단, 회의는 한번 시작하면 중간에 중단될 수 없으며 한 회의가 끝나는 것과 동시에 다음 회의가 시작될 수 있다. 회의의 시작시간과 끝나는 시간이 같을 수도 있다. 이 경우에는 시작하자마자 끝나는 것으로 생각하면 된다.

입력

첫째 줄에 회의의 수 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N+1 줄까지 각 회의의 정보가 주어지는데 이것은 공백을 사이에 두고 회의의 시작시간과 끝나는 시간이 주어진다. 시작 시간과 끝나는 시간은 231-1보다 작거나 같은 자연수 또는 0이다.

출력

첫째 줄에 최대 사용할 수 있는 회의의 최대 개수를 출력한다.

코드

#include <algorithm>
#include <cstring>
#include <iostream>
#include <map>
#include <set>
#include <stack>
#include <vector>

using namespace std; 

int N, K;
vector<pair<int,int>> A;
pair<int, int> m;
int ans = 1;

bool cmp(pair<int,int> a, pair<int,int> b){
	if(a.second == b.second){
		return a.first < b.first;
	}
	else{
		return a.second < b.second;
	}
}

int main(void){
	cin >> N;
	for(int i=0; i<N; ++i){
		int a, b;
		cin >> a >> b;
		m.first = a;
		m.second = b;
		A.push_back(m);
	}

	sort(A.begin(), A.end(), cmp);

	int start = 0;
	bool isNext = false;
	while(start < N-1){
		isNext = false;
		for(int i=start+1; i<N; ++i){
			if(A[i].first >= A[start].second){
				start = i;
				isNext = true;
				ans++;
				break;
			}
		}

		if(!isNext){
			break;
		}
	}

	cout << ans;
	
	return 0;
}

이 문제도 전 문제 동전0과 같이 그리디 알고리즘의 대표적인 문제이다.

그리디 알고리즘을 활용할 땐 sort가 유용하게 쓰인다. 최소값부터 쓸일이 많은데 sort를 활용하면 이걸 잘 사용할수 있다.

여기선 먼저 끝나는 시간이 젤 빠른순대로 정렬하고 시작시간과 끝나는 시간이 겹치지 않는 회의들의 갯수를 정렬 한 순으로 세주기만 하면 답이 나온다.

sort에서 쓰이는 cmp함수 작성 방법을 숙지해놓자.(etc. vector안에 pair가 있는 경우 정렬 방법)