백준 1931번 - 회의실 배정(C++/greedy)

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

1931번: 회의실 배정

문제

한 개의 회의실이 있는데 이를 사용하고자 하는 N개의 회의에 대하여 회의실 사용표를 만들려고 한다. 각 회의 I에 대해 시작시간과 끝나는 시간이 주어져 있고, 각 회의가 겹치지 않게 하면서 회의실을 사용할 수 있는 회의의 최대 개수를 찾아보자. 단, 회의는 한번 시작하면 중간에 중단될 수 없으며 한 회의가 끝나는 것과 동시에 다음 회의가 시작될 수 있다. 회의의 시작시간과 끝나는 시간이 같을 수도 있다. 이 경우에는 시작하자마자 끝나는 것으로 생각하면 된다.
입력첫째 줄에 회의의 수 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N+1 줄까지 각 회의의 정보가 주어지는데 이것은 공백을 사이에 두고 회의의 시작시간과 끝나는 시간이 주어진다. 시작 시간과 끝나는 시간은 231-1보다 작거나 같은 자연수 또는 0이다.

접근법

끝나는 시간을 기준으로 정렬을 한다.

즉 시작시간과 끝나는 시간을 pair에 저장한다 가정하면,

pair.second를 기준으로 정렬해야한다. 

다만 4->4, 3->4 처럼 끝나는 시간은 같지만 시작 시간이 다른 경우에도 정렬 해줘야 한다

이거 때문에 시간을 좀 썼다.

 

끝나는 시간으로 정렬을 했다면, 끝나는 시간을 0으로 잡고, 빠른것부터 넣어준다.

즉 정렬된 것이 (1,2)(1,3)(2,4)(3,6) 이라면

먼저 (1,2)를 넣어준다. (cnt ++)

이전의 끝나는 시간이 (위 예제로 하면 2)

startTime보다 작거나 같아진다면 (위 예제로하면 2,4)

끝나는 시간을 4로 바꾸어 주고, cnt++를 해준다.

이를 반복해주면 된다.

 

위에서 말한대로 정렬을 해주었다면 문제는 없다.

 

다만 나는 24시간 기준인줄 알고 배열을 24까지만 만들어 두었는데 그게 아니다. 하루를 기준으로 시간표를 짜는것이 아닌, 2^31-1 시간만큼의 범위에서 시간표를 짜는것이다

코드

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

using namespace std;

int main()
{
    vector<pair<int, int>> time;
    int N;
    cin >> N;
    for (int i = 0; i < N; i++)
    {
        int from, to;
        cin >> from >> to;
        time.push_back(make_pair(from, to));
    }

    sort(time.begin(), time.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 cnt = 0;
    int endTime = 0;
    for (int i = 0; i < time.size(); i++)
    {
        if (time[i].first >= endTime)
        {
            endTime = time[i].second;
            cnt++;
        }
    }
    cout << cnt;
    return 0;
}

시간이 조금 걸렸다 ㅎㅎ. 

문제 파악하는 것 보단, 알고리즘 반례를 찾아내느라 시간이 오래걸렸다.