LeetCode 1942. 最小未被占据椅子的编号（set）

示例 1：
输入：times = [[1,4],[2,3],[4,6]], targetFriend = 1
输出：1
解释：
- 朋友 0 时刻 1 到达，占据椅子 0 。
- 朋友 1 时刻 2 到达，占据椅子 1 。
- 朋友 1 时刻 3 离开，椅子 1 变成未占据。
- 朋友 0 时刻 4 离开，椅子 0 变成未占据。
- 朋友 2 时刻 4 到达，占据椅子 0 。
朋友 1 占据椅子 1 ，所以返回 1 。

示例 2：
输入：times = [[3,10],[1,5],[2,6]], targetFriend = 0
输出：2
解释：
- 朋友 1 时刻 1 到达，占据椅子 0 。
- 朋友 2 时刻 2 到达，占据椅子 1 。
- 朋友 0 时刻 3 到达，占据椅子 2 。
- 朋友 1 时刻 5 离开，椅子 0 变成未占据。
- 朋友 2 时刻 6 离开，椅子 1 变成未占据。
- 朋友 0 时刻 10 离开，椅子 2 变成未占据。
朋友 0 占据椅子 2 ，所以返回 2 。
 
提示：
n == times.length
2 <= n <= 10^4
times[i].length == 2
1 <= arrivali < leavingi <= 10^5
0 <= targetFriend <= n - 1
每个 arrivali 时刻 互不相同 。

class Solution {
public:
    int smallestChair(vector<vector<int>>& times, int targetFriend) {
        int n = times.size();
        vector<int> id(n);
        iota(id.begin(), id.end(), 0); // 编号 0 - n-1
        sort(id.begin(), id.end(), [&](auto a, auto b){
            if(times[a][0] == times[b][0])
                return times[a][1] < times[b][1];
            return times[a][0] < times[b][0];
        }); // 排序先来的先取号，先走的，先放回号码
        set<int> untake(id.begin(), id.end()); // 没拿的号码
        set<pair<int,int>> leaveTime; // 离开时间，拿的号码，自动排序，时间早的先离开
        for(auto idx : id)
        {
            while(!leaveTime.empty() && (*leaveTime.begin()).first <= times[idx][0]) 
            {   // 到达之前，有人离开
                int num = (*leaveTime.begin()).second;//拿的号码
                untake.insert(num);//放回号码
                leaveTime.erase(leaveTime.begin());//离开了
            }
            if(idx == targetFriend)
                return *untake.begin();
            int takeNum = *untake.begin();//分配号码
            untake.erase(takeNum);
            leaveTime.insert({times[idx][1], takeNum});
        }
        return -1;
    }
};