leetcode 5736 周赛 单线程cpu 优先队列和排序

 通过这个题熟悉了下iota的用法，vector自定义排序（根据另一个数组来排当前的数组） 优先队列对pair数据的处理方式，很好的一道题

class Solution {
public:
    using PII = pair<int,int>;//type def 

    vector<int> getOrder(vector<vector<int>>& tasks) {
        int size = tasks.size(),time = 0,pos = 0;//pos 对tasks按照开始执行时间排序，保存其index，pos对index数组遍历，然后从tasks中依次取出元素
        vector<int> ret;//存储结果的vector
        vector<int> v(size,0);//存储index的
        iota(v.begin(),v.end(),0);//1,2,3,4...size
        sort(v.begin(),v.end(),[&](int a,int b){return tasks[a][0] < tasks[b][0];});//按开始执行时间排序index
        priority_queue<PII,vector<PII>,greater<PII>> q;//最小堆

        while(pos<size)
        {
            if(q.empty())//队列为空 在当前时间点内无任务到达 就直接去跳到下一个任务
                time = max(tasks[v[pos]][0],time);
            while(pos<=size-1 && time >= tasks[v[pos]][0])
                q.emplace(tasks[v[pos]][1],v[pos++]);//入堆 
                //对于pair，堆会先比较第一个元素，再比较第二个元素，既满足了执行时间短的先执行也满足了时间相等时下标小的先执行
            time += q.top().first;
            ret.emplace_back(q.top().second);//存结果
            q.pop();
            //if(ret.size() == size)  break;//也可以等到全入堆后就break，然后依次pop
        }
        while(!q.empty())
        {
            ret.emplace_back(q.top().second);
            q.pop();
        }
        return ret;
    }
};

自定义函数也可以不用那么sao：还是常规写法看着舒服点

iota的用法