问题描述
我们需要重新排列队列,满足任何一个"person
"中h和k的含义。
解法探究
每一个"person"的k值实际上只与h值大于等于它的"person"有关,单独以某一个"person"为参照,除去队列中所有h值比它小的元素,其k值就是它在队列中的位置序号。那么我们不妨按照h值降序处理元素,先找出最高的那群人,以k值为角标进行排列,再找出次高的人群,按照其各自的k值插入到新队列中,如此执行直到将原队列中的所有元素都遍历过。代码实现如下:
class Solution {
public:
vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
sort(people.begin(), people.end(),[](pair<int,int> p1, pair<int,int> p2){
return p1.first > p2.first || (p1.first == p2.first && p1.second < p2.second);
});
vector<pair<int,int>> sol;
for (auto person : people){
sol.insert(sol.begin() + person.second, person);
}
return sol;
}
};
这里先用库函数sort对people进行了排序,再将所有元素逐个插入新队列,代码十分简洁有力。