LeetCode第 406 题：根据身高重建队列(C++)

406. 根据身高重建队列 - 力扣（LeetCode）

看标签是贪心算法的思路，可是为啥贪心就合理呢？虽然是中等难度，可是想好久没想出为啥…

实际和贪心没啥关系。。。

首先是先排序，按照第一个元素逆序，第二个元素正序，那么每一个元素之前的元素的个数，就是大于等于它的元素数量。个头高的人会忽略个头比它矮的人，只会在意个头大于等于它的人。

样例：

[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]

排序之后：

[[7,0], [7,1], [6,1], [5,0], [5,2], [4,4]]

下面一步一步构造结果数组res，怎么构建呢？对于每一个元素 v，可以认为 v 之前的元素（高度 >= v[0]）都已经处理完并且存入res，那么res.size()就是高度 >= v[0]的元素个数。而元素 v 的存储位置要求高度 >= v[0]的元素个数等于v[1]，那我们就只需要从左往右数，数到第 v[1] 个位置的时候，将元素 v 插入即可。

这样的插入方式不会影响后面的元素，举个例子：

假设元素v = [6,1]，元素[7,0], [7,1]已经处理好，也就是当前res为：

[[7, 0], [7, 1]]

那么v[1] == 1，左往右数，下标为 1 处将元素 v 插入：

[[7, 0], [6, 1], [7, 1]]

现在元素 v 的位置满足了要求，而且插入位置后面的元素[7, 1]也不会收到影响，因为元素[7, 1]是看不到比它矮的元素[6, 1]的。

那么只需要按照这个逻辑将元素挨个插入即可：

更好的解释可以看这位同学做的ppt:【先排序，再插队】动画演示算法过程，有点小套路 - 根据身高重建队列 - 力扣（LeetCode）

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(), people.end(), [](const vector<int> &a, const vector<int> &b){
            return a[0] > b[0] || (a[0] == b[0] && a[1] < b[1]);
        });//第一个元素逆序，第二个元素正序
        //在某元素之前的元素个数（下面的res.size()），就是大于等于它的元素的数量
        vector<vector<int>> res;
        for(const auto &v : people)
            res.insert(res.begin()+v[1], v);
        return std::move(res);
    }
};

使用list优化插入效率：

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        std::sort(people.begin(), people.end(), [](const vector<int>& a, const vector<int>& b){
            return a[0] > b[0] || (a[0] == b[0] && a[1] < b[1]);
        });
        list<vector<int>> l;//方便插入
        for (auto& v : people) {
            auto it = l.begin();
            advance(it, v[1]);//将迭代器it向前移动v[1]个位置
            l.insert(it, v);
        }
        return {l.begin(), l.end()};
    }
};

换一种类似拓扑排序的思路：

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        //拓扑排序的思路
        //每次选择v[1] == 0(无需前面有大于等于它元素)且v[0]最小的元素v插入
        //然后将其他第一个值 < v[0] 的元素的第二个值-1(表示它前面已经有一个元素了)
        vector<vector<int>> res, backup(people);//需要备份
        int n = people.size();
        for(int i = 0; i < n; ++i){
            int idx = -1;
            for(int j = 0; j < n; ++j){
                if(people[j][1] == 0){
                    if(idx == -1)   idx = j;
                    else idx = people[j][0] < people[idx][0] ? j : idx; 
                }
            }
            res.push_back(backup[idx]);//使用的是备份，因为people里的值已经被修改了
            for(auto &v : people)
                if(v[0] <= people[idx][0])  --v[1];
        }
        return res;
    }
};