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

思路

首先我们将输入排下序
[h,k]
[
[7,0],
[7,1],
[6,1],
[5,0],
[5,2],
[4,4],
]
k是排在这个人前面且身高>=这个人的人数，我们观察一下最后一个人，[4,4]，他的身高是4所有人中最矮的，k为4，这个人前面身高大于等于他的人有四个，由于他是最矮的，所以前面也不可能有身高小于他的，所以，这个人就排在索引为4的位置。
同理，[5,2]在前5个人中也是最小，所以他在前5个人中排在索引为2的位置。
[5,0]在前5个人中位于索引0
[6,1]在前三个人中位于索引1
[7,1]在前两个人中位于索引1
[7,0]在前两个人中位于索引0

我们反过来思考，我们先排最大的身高，排好后为[[7,0],[7,1]]，然后将后面的加进来则为[[7,0],[6,1],[7,1]],循环这个过程最后得到[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]

代码：

bool cmp(vector<int> &v1,vector<int> &v2){
    if(v1[0]!=v2[0])return v1[0]>v2[0];
    return v1[1]<v2[1];
}

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(),people.end(),cmp);
        vector<vector<int>> res;
        for(auto &p:people){
            res.insert(res.begin()+p[1],p);
        }
        return res;
    }
};