【LeetCode-406】根据身高重建队列

问题

假设有打乱顺序的一群人站成一个队列，数组 people 表示队列中一些人的属性（不一定按顺序）。每个 people[i] = [hi, ki] 表示第 i 个人的身高为 hi ，前面 正好 有 ki 个身高大于或等于 hi 的人。

请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ，其中 queue[j] = [hj, kj] 是队列中第 j 个人的属性（queue[0] 是排在队列前面的人）。

示例

输入： people = [[6,0],[5,0],[4,0],[3,2],[2,2],[1,4]]
输出： [[4,0],[5,0],[2,2],[3,2],[1,4],[6,0]]

解答

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(), people.end(), cmp);
        list<vector<int>> ls;
        for (auto& v : people) {
            auto it = ls.begin();
            for (int i = 0; i < v[1]; i++) it++;
            ls.insert(it, v);
        }
        return vector<vector<int>>(ls.begin(), ls.end());
    }
private:
    static bool cmp(vector<int>& a, vector<int>& b) {
        if (a[0] == b[0]) return a[1] < b[1];
        return a[0] > b[0];
    }
};

重点思路

本题需要注意的是链表STL的使用以及类中自定义sort函数时需要加static关键字。

首先对输入数组进行排序，具体方法为：身高降序，如果身高一样，则按“排在前面的身高不比自身低的人数”升序。理由为：设a身高比b高，当a的位置确定后，无论b插在a的前面还是后面，都不会影响a所在位置的正确性，这是身高升序的原因。而第二项升序排列，是因为希望前面人数多的后面再插入，方便list的插入操作。