根据身高重建队列
题目描述
假设有打乱顺序的一群人站成一个队列,数组 people 表示队列中一些人的属性(不一定按顺序)。每个 people[i] = [hi, ki] 表示第 i 个人的身高为 hi ,前面 正好 有 ki 个身高大于或等于 hi 的人。
请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ,其中 queue[j] = [hj, kj] 是队列中第 j 个人的属性(queue[0] 是排在队列前面的人)。
解题思路
首先我们根据身高进行排序,高的在前面,如果一样高,前面比自己高的人少的在前面。
然后我们遍历这个从高到低的序列,每次每个人他前面有 k 个比他高的人,然后放到第 k 个位置,因为后面再插入到他前面也不会影响到他的这个数据了。
遍历完就是答案。
题解
class Solution {
public:
bool static cmp(vector<int> a1, vector<int> a2){
if(a1[0] > a2[0]){
return true;
}
else if(a1[0] < a2[0]){
return false;
}
else{
return a1[1] < a2[1];
}
}
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
int n = people.size();
vector<vector<int>> ans;
sort(people.begin(), people.end(), cmp);
if(int i = 0; i < n; i++){
int pos = people[i][1];
ans.insert(ans.begin() + pos, people[i]);
}
return ans;
}
};