2402 meeting rooms 3

[kitano-kazuki]

https://leetcode.com/problems/meeting-rooms-iii/description

[nodchip]

自分も解いてみました。自分も遅延された予約がある場合の処理が複雑になりそうだと思いました。結局、遅延された予約の処理をしない方向で考え、ミーティングルームが空いていなかったら、空くまで待ってからミーティングを開始するという方向に落ち着きました。どちらが良いかは好みの範囲だと思います。

class Solution {
public:
    int mostBooked(int n, vector<vector<int>>& meetings) {
        std::sort(meetings.begin(), meetings.end());
        std::vector<int> num_meetings(n);
        std::set<int> available_meeting_rooms;
        for (int i = 0; i < n; ++i) {
            available_meeting_rooms.insert(i);
        }
        // 終了時刻, 部屋番号
        std::priority_queue<std::pair<int64_t, int>, std::vector<std::pair<int64_t, int>>, std::greater<std::pair<int64_t, int>>> ongoing_meetings;
        for (const vector<int>& meeting : meetings) {
            int64_t start = meeting[0];
            int64_t end = meeting[1];

            // 終了したミーティングを取り除く。
            while (!ongoing_meetings.empty() && ongoing_meetings.top().first <= start) {
                int room = ongoing_meetings.top().second;
                ongoing_meetings.pop();
                available_meeting_rooms.insert(room);
            }

            // 部屋が空いていたらミーティングを開始する。
            if (!available_meeting_rooms.empty()) {
                int room = *available_meeting_rooms.begin();
                available_meeting_rooms.erase(available_meeting_rooms.begin());
                ++num_meetings[room];
                ongoing_meetings.push(std::make_pair(end, room));
                continue;
            }

            // 部屋が空いていなかったら、空くまで待ってミーティングを開始する。
            int64_t last_end = ongoing_meetings.top().first;
            int room = ongoing_meetings.top().second;
            ongoing_meetings.pop();
            int64_t new_end = end - start + max(last_end, start);
            ++num_meetings[room];
            ongoing_meetings.push(std::make_pair(new_end, room));
        }

        int max_meetings = *std::max_element(num_meetings.begin(), num_meetings.end());
        return std::distance(num_meetings.begin(), std::find(num_meetings.begin(), num_meetings.end(), max_meetings));
    }
};

[kitano-kazuki]

解いていただきありがとうございます。

空くまで待ってからミーティングを開始するという方向

これは思いつかなかったです。
直近で解放される時刻に今処理中の予約の所要時間を足して終了予定時刻として, ongoing_meetingsに追加するということですね。

予約の確保を遅延しないので, 処理を追うのが楽で自分もいいと思いました。