leetcode406

最新推荐文章于 2024-08-24 19:54:04 发布

原创最新推荐文章于 2024-08-24 19:54:04 发布 · 701 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

429 篇文章

订阅专栏

本文介绍了一种使用STL实现的队列重构算法，通过贪婪策略对pair元素进行排序，以解决特定问题。文章详细阐述了排序逻辑及其实现细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

并不是太简单，还是有点难度，这是stl的胜利。

本题采用greedy的思路，对于pair(h,k)。首先确定优先顺序：按照k从小到大的顺序排序；当k相同时，按照h从小到大排序。

首先 static bool cmp函数用于sort内使用，必须加上static，不然会出错。

std::sort(people.begin(),people.end(),cmp)对people排序。

distance函数可以返回正值或负值，根据情况而定，就是返回distance(begin,end)——-(end-begin).

itOld=result.insert(it,people[i]);返回插入后的地址。

每次操作前都需要比较一番插入前后的地址。

class Solution {
public:

    static bool cmp( pair<int,int>& a,  pair<int,int>& b)
    {
        if(a.second<b.second)
            return true;
        else if(a.second==b.second&&a.first<b.first)
            return true;
        else
            return false;
    }

    vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
        if(people.size()==0)
            return people;
        else
        {
            std::sort(people.begin(),people.end(),cmp);

            vector<pair<int, int>> result;
            result.push_back(people[0]);
            vector<pair<int, int>>::iterator it=result.begin();
            vector<pair<int, int>>::iterator itOld=result.begin();

            for(int i=1;i<people.size();i++)
            {
                int num=people[i].first;
                int time=people[i].second;

                int j=0;
                while(time!=0)
                {
                    if(result[j].first>=num)
                        time--;
                    j++;
                }

                it=result.begin()+j;
                if(people[i-1].second==people[i].second)
                {
                    if(distance(itOld,it)<=0)
                        it=itOld+1;
                }
                itOld=result.insert(it,people[i]);
            }
            return result;
        }
    }
};