406.根据身高重建队列-贪心算法

题目描述

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

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

示例 1：

输入：people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]
输出：[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]
解释：
编号为 0 的人身高为 5 ，没有身高更高或者相同的人排在他前面。
编号为 1 的人身高为 7 ，没有身高更高或者相同的人排在他前面。
编号为 2 的人身高为 5 ，有 2 个身高更高或者相同的人排在他前面，即编号为 0 和 1 的人。
编号为 3 的人身高为 6 ，有 1 个身高更高或者相同的人排在他前面，即编号为 1 的人。
编号为 4 的人身高为 4 ，有 4 个身高更高或者相同的人排在他前面，即编号为 0、1、2、3 的人。
编号为 5 的人身高为 7 ，有 1 个身高更高或者相同的人排在他前面，即编号为 1 的人。
因此 [[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]] 是重新构造后的队列。

思路

本题有两个维度，h和k，看到这种题目一定要想如何确定一个维度，然后再按照另一个维度重新排列。
首先找规律，h相同的，k越小位置越靠前；对于h很小的，位置基本取决于k值，因为前面都比它高。
第一次重排可以按h从高到低进行排列，相同身高k小的在前。
第二次重排就要按k从高到低进行插入，k值就是此人要插入的位置。由于k值是身高大于或等于 hi 的人，所以并不影响前面高个子的站序依然满足要求。

C++代码

语法细节

第一次排序可以用sort对h排序，然后再遍历一次将相同身高的依据k进行调整。
但是有一种更高级的方法，就是自定义sort的排序准则。

static bool cmp(vector<int>& x, vector<int>& y){
        if(x[1] == y[1]){
            return x[0] < y[0];
        }
        return x[0] > y[0];
    }

第二次排序用到数组的插入，C++的数组经常用vector或list实现，写的时候没想起来插入的函数：insert(iterator p, Element)。
当然这两种都能实现，但使用vector是非常费时的，C++中vector（可以理解是一个动态数组，底层是普通数组实现的）如果插入元素大于预先普通数组大小，vector底部会有一个扩容的操作，即申请两倍于原先普通数组的大小，然后把数据拷贝到另一个更大的数组上。
所以使用vector（动态数组）来insert，是费时的，插入再拷贝的话，单纯一个插入的操作就是O(n²)了，甚至可能拷贝好几次，就不止O(n²)了。

vector实现(执行用时53ms)：

class Solution {
public:
    static bool cmp(vector<int>& x, vector<int>& y){
        if(x[0] == y[0]){
            return x[1] < y[1];
        }
        return x[0] > y[0];
    }
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        //用vector实现
        sort(people.begin(),people.end(),cmp);  //按h从高到低排列
        vector<vector<int>> que;
        for(auto it = people.begin(); it != people.end(); it ++){
            int position = (*it)[1];
            que.insert(que.begin() + position, (*it));
        }
        return que;
    }
};

List实现（执行用时25ms）

class Solution {
public:
    static bool cmp(vector<int>& x, vector<int>& y){
        if(x[0] == y[0]){
            return x[1] < y[1];
        }
        return x[0] > y[0];
    }
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {       
        sort(people.begin(),people.end(),cmp);  //按h从高到低排列
        //用list实现
        list<vector<int>> que; // list底层是链表实现，插入效率比vector高的多
        for (int i = 0; i < people.size(); i++) {
            int position = people[i][1]; // 插入到下标为position的位置
            std::list<vector<int>>::iterator it = que.begin();
            while (position--) { // 寻找在插入位置
                it++;
            }
            que.insert(it, people[i]);
        }
        return vector<vector<int>>(que.begin(), que.end());
    }
};

JAVA代码

待补充（学完JAVA中list的用法再来练习）