算法设计周记（十四）--Queue Reconstruction by Height

问题描述

我们需要重新排列队列，满足任何一个"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进行了排序，再将所有元素逐个插入新队列，代码十分简洁有力。