贪心算法+容器list+advance 406. 根据身高重建队列

最新推荐文章于 2023-05-23 10:31:04 发布

豌豆射手GCC

最新推荐文章于 2023-05-23 10:31:04 发布

阅读量293

点赞数

分类专栏：面向对象C++ leetcode 贪心算法

本文链接：https://blog.csdn.net/BLUEsang/article/details/106278498

版权

leetcode 同时被 3 个专栏收录

189 篇文章 0 订阅

订阅专栏

面向对象C++

34 篇文章 5 订阅

订阅专栏

贪心算法

15 篇文章 0 订阅

订阅专栏

406. 根据身高重建队列

假设有打乱顺序的一群人站成一个队列。每个人由一个整数对(h, k)表示，其中h是这个人的身高，k是排在这个人前面且身高大于或等于h的人数。编写一个算法来重建这个队列。

注意：
总人数少于1100人。

示例

输入:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]

输出:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]

解题
先对people进行排序——h从大小，若相等，则k从小到大；
按顺序将people中每个元素插入list容器下标为k的位置；
将list容器转换为int，传出即可；

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
          sort(people.begin(), people.end(),
                [](const vector<int>& lhs, const vector<int>& rhs)
                 {return lhs[0] == rhs[0] ? lhs[1] <= rhs[1] : lhs[0] > rhs[0];});
        //先按h降序，再按k升序
        int n=people.size();
        list<vector<int>> tmp;
        for(int i=0;i<n;i++){
            auto pos=tmp.begin();
            advance(pos,people[i][1]);
            tmp.insert(pos,people[i]);
        }
        return vector<vector<int>> (tmp.begin(),tmp.end());
    }
private:    
    vector<vector<int>> res;
};