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