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