【贪心】406. Queue Reconstruction by Height

云水谣CS

于 2018-07-21 17:15:18 发布

阅读量179

点赞数

分类专栏： LeetCode reView

本文链接：https://blog.youkuaiyun.com/leetcodecl/article/details/81146903

版权

LeetCode 同时被 2 个专栏收录

107 篇文章

订阅专栏

reView

9 篇文章

订阅专栏

本文介绍了一种使用排序和插入操作来解决队列重建问题的算法实现。通过C++和Python两种语言展示了如何根据个体的高度和视线前的人数来正确重建队列的方法。

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

C++版：

class Solution {
public:
    static bool cmp(pair<int,int> A, pair<int,int> B)
    {
        if(A.first!=B.first)
            return A.first>B.first;
        else
            return A.second<B.second;
    }
    vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
        sort(people.begin(),people.end(),cmp);
        vector<pair<int,int>> ans;
        for(auto item : people)
        {
            ans.insert(ans.begin()+item.second,item);
        }
        return ans;
    }
};

python3版：

class Solution:
    def reconstructQueue(self, people):
        """
        :type people: List[List[int]]
        :rtype: List[List[int]]
        """
        def cmpa(A):
            return -A[0],A[1]
        people.sort(key=cmpa)
        ans = []
        for item in people:
            ans.insert(item[1],item)
        return ans