[알고리즘 풀이 분석] BOJ 1931 회의실 배정

어제의 연장선으로 오늘도 그리디 문제를 하나 풀어보았다. 실버 2 단계로 어렵지는 않은 간단한 그리디 문제이고 문제는 BOJ 1931 회의실 배정 이다.

BOJ 1931 회의실 배정

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

입출력

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

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

나의 풀이

어제 풀어본 문제와는 조금 다르게 하나의 회의실에 최대한 많은 회의가 진행될 수 있도록 하는 문제이다.

처음엔 회의 시작 시간을 기준으로 정렬했지만 최대한 많은 회의가 회의실에 들어가야 하므로 회의 종료 시간이 빠를수록 좋다는 것을 깨달았다.

이때 회의 종료 시간은 항상 시작 시간보다 늦거나 같으므로 시작 시간을 따로 고려할 필요는 없지만, 회의 종료 시간이 같은 경우라면 시작 시간이 빠른 것이 먼저 배정될 수 있도록 해서 회의실을 사용할 수 있는데 비워두지 않도록 해야 했다.

정렬된 회의들을 탐색하면서 회의실에 들어가 있는 회의의 종료 시간 endTime 보다 현재 회의의 시작 시간이 늦거나 같으면 배정하고 그렇지 않으면 다음 회의로 탐색을 진행한다.

코드

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

// boj 1931 회의실 배정, 실버 2, 그리디
using namespace std;

bool byEnd(pair<int, int> a, pair<int, int> b){  // 강의 시작 시간 기준 오름 차순 정렬
    if (a.second==b.second) return a.first<b.first;
    return a.second < b.second;
}

int getMaxMeeting(int N, vector<pair<int, int>> meeting){
    sort(meeting.begin(), meeting.end(), byEnd);

    int max_meeting = 1, i = 1;
    int endTime = meeting[0].second;

    while (i<N){
        if (endTime <= meeting[i].first){
            max_meeting++;
            endTime = meeting[i].second;
        }
        i++;
    }

    return max_meeting;
}

int main(){

    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int N;
    cin>>N;

    vector<pair<int, int>> meeting(N, {-1,-1});
    for (int i = 0; i < N; ++i) {
        cin>>meeting[i].first>>meeting[i].second;
    }

    cout<<getMaxMeeting(N, meeting);

    return 0;
}