leetcode 406. 根据身高重建队列（C++、python）

假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]

思路：首先按身高从高到矮进行排序。然后利用insert函数，依次进行插入操作。

例如，当前元素为[25,3]，已经插入的元素有5个（身高肯定大于等于25），那么，就在索引是3的位置输入[25,3]，依次类推

C++

class Solution {
public:
    static bool cmp(vector<int>& a, vector<int>& b)
    {
        if(a[0]==b[0])
        {
            return a[1]<b[1];
        }
        else
        {
            return a[0]>b[0];
        }
    }
    
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) 
    {
        int n=people.size();
        sort(people.begin(),people.end(),cmp);
        vector<vector<int>> res;
        for(int i=0;i<n;i++)
        {
            res.insert(res.begin()+people[i][1],people[i]);
        }
        return res;
    }
};

python

class Solution:
    def reconstructQueue(self, people: List[List[int]]) -> List[List[int]]:
        n=len(people)
        people=sorted(people,key=lambda x:(-x[0],x[1]))
        res=[]
        for s in people:
            res.insert(s[1],s)
        return res