[Baekjoon] #1931 - 회의실 배정

문제

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

입력

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

출력

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

---

풀이

풀이과정

나도 모르게 문제 페이지 하단의 알고리즘 분류에서 그리디 알고리즘이라는 것을 봐버렸다..
그래서 그냥 그리디 알고리즘으로 풀기로 했다.
생각해보니까 이게 맞는 것 같아서도 있다.

1. 회의들을 '끝나는 시간'을 기준으로 오름차순 정렬한다.
2. 그리디 알고리즘으로, 다음으로 가능한 회의 중 가장 빨리 끝나는 것을 선택한다.
3. 위 과정을 반복하여 모든 회의들을 순회한다.

처음에 이렇게만 했는데, 85%에서 틀렸다고 떴다.
그 이유는 끝나는 시간으로만 정렬된 상태여서, 예를 들어서

//ex
3
2 2
1 2
2 3

와 같은 경우에는 '2 2' 전에 '1 2'를 넣을 수 있음에도 넣지 않게 되기 때문이었다.
그래서 끝나는 시간이 동일한 경우에는 시작 시간을 기준으로 정렬되게끔 추가하니, 통과되었다.

최종 코드

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

using namespace std;

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

    int N;
    cin >> N;

    vector<pair<int, int>> meetings;
    for(int i=0; i<N; i++){
        int start, end;
        cin >> start >> end;
        meetings.push_back({start, end});
    }

    //끝나는 시간을 기준으로, 오름차순으로 정렬(끝나는 시간이 같으면 시작 시작 기준으로 추가 정렬)
    sort(meetings.begin(), meetings.end(), [](const pair<int, int>& a, const pair<int, int>& b) {
        if (a.second == b.second)
            return a.first < b.first; // 끝나는 시간이 같으면, 시작 시간 오름차순
        return a.second < b.second; // 끝나는 시간 오름차순
    });
    

    int count = 1;
    int curStart = meetings[0].first; int curEnd = meetings[0].second;
    for(int i=1; i<N; i++){
        if(meetings[i].first >= curEnd){
            curStart = meetings[i].first;
            curEnd = meetings[i].second;
            count++;
        }
    }
    cout << count;

    return 0;
}

새로 알게된 점

• sort() 함수에서, 마지막으로 보내는 정렬 기준이 되는 함수를 보면 const pair<int, int>& a에서 그냥 a라고 하지 않고 &를 붙여준 것을 볼 수 있다.
  이렇게 참조로 변수값을 사용하면 변수를 복사하지 않기 때문에 메모리도 절약할 수 있어 효율적이다.