题目描述
假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]
问题分析
先给数组排序,按照身高从高到低排,如果身高相同,则第二个数小的排前面。接下来我们从前往后遍历排序后的数组people,然后根据遍历到的元素的第二个数,将其插入到ans数组中对应的位置。遍历结束后返回ans。这里解释一下:因为第二个数k表示排在这个人前面且身高大于或等于h
的人数,并且我们是以身高降序的顺序插入到数组ans中,此时在数组ans中的人的身高都大于等于当前待插入的人,所以当前元素的第二个数就是要在ans中插入的位置。
代码实现
class Solution {
public:
static bool cmp(vector<int>& v1, vector<int>& v2){
if(v1[0] == v2[0])
return v1[1] < v2[1];
return v1[0] > v2[0];
}
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(), people.end(), cmp);
vector<vector<int>> ans;
for(auto a : people){
ans.insert(ans.begin() + a[1], a);
}
return ans;
}
};