Leetcode 406. 根据身高重建队列

题目

Leetcode 406. 根据身高重建队列

代码（7.25 首刷看解析）

做这个题的时候，要先确定一个维度，再确定另一个维度。如果两个维度一起考虑一定会顾此失彼。

这里先确定身高h，再确定k。

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        auto cmp = [&](vector<int>& a, vector<int>& b){
            if(a[0] == b[0]) return a[1] < b[1];
            return a[0] > b[0];
        };
        sort(people.begin(), people.end(), cmp);
        vector<vector<int>> res;
        for(auto& p : people) {
            int position = p[1];
            res.insert(res.begin() + position, p);
        }
        return res;
    }
};

但使用vector是非常费时的，C++中vector（可以理解是一个动态数组，底层是普通数组实现的）如果插入元素大于预先普通数组大小，vector底部会有一个扩容的操作，即申请两倍于原先普通数组的大小，然后把数据拷贝到另一个更大的数组上。

所以使用vector（动态数组）来insert，是费时的，插入再拷贝的话，单纯一个插入的操作就是O(n²)了，甚至可能拷贝好几次，就不止O(n²)了。

改用list，注意，iterator是只能进行++操作的。

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        auto cmp = [&](vector<int>& a, vector<int>& b){
            if(a[0] == b[0]) return a[1] < b[1];
            return a[0] > b[0];
        };
        sort(people.begin(), people.end(), cmp);
        list<vector<int>> res;
        for(auto& p : people) {
            int position = p[1];
            auto it = res.begin();;
            while(position--) it++;
            res.insert(it, p);
        }
        return vector<vector<int>>(res.begin(), res.end());
    }
};

代码（10.1 二刷看解析）

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(), people.end(), [](auto& a, auto& b) {
            if(a[0] == b[0])
                return a[1] < b[1];
            return a[0] > b[0];
        });
        // vector<vector<int>> res;
        list<vector<int>> res;
        for(auto& p : people) {
            int k = p[1];
            auto it = res.begin();
            while(k--) {
                it++;
            }
            res.insert(it, p);
        }
        return vector<vector<int>>(res.begin(), res.end());
    }
};