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]]

 

思路

首先因为第二维只记录了当前大于等于人身高的数，所以身高高的人是看不见身高比自己低的，假设乱顺序队列如下

【（170，2），（180，0）， （190，0）】这里面身高最高的是190，其看不见身高比自己低的，所以其不会影响比自己低的人，设排好队队列【】，则先将最高身高插入队列，插到的位置根据其前面比自己高的人来确定，因为前面没有比自己高的，所以直接插到0位置【（190，0）】；然后将第二高的人插入队列，其前面也没有比自己高的，所以也直接插到0位置【（180，0）， （190，0）】；最后将170插入队列，因为这个队列已经是有序的了，比170高的有两个，所以直接插到第二个位置即可。

所以，先对数组按照身高从高到低排序，如果身高一样，则按照第二维大小从小到大排序。遍历数组，根据第二维大小将人插到顺序队列中。

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

 

代码