算法-贪心篇11-根据身高重建队列

根据身高重建队列

力扣题目链接

题目描述

假设有打乱顺序的一群人站成一个队列，数组 people 表示队列中一些人的属性（不一定按顺序）。每个 people[i] = [hi, ki] 表示第 i 个人的身高为 hi ，前面 正好 有 ki 个身高大于或等于 hi 的人。
请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ，其中 queue[j] = [hj, kj] 是队列中第 j 个人的属性（queue[0] 是排在队列前面的人）。

解题思路

首先我们根据身高进行排序，高的在前面，如果一样高，前面比自己高的人少的在前面。
然后我们遍历这个从高到低的序列，每次每个人他前面有 k 个比他高的人，然后放到第 k 个位置，因为后面再插入到他前面也不会影响到他的这个数据了。
遍历完就是答案。

题解

class Solution {
public:
    bool static cmp(vector<int> a1, vector<int> a2){
        if(a1[0] > a2[0]){
            return true;
        }
        else if(a1[0] < a2[0]){
            return false;
        }
        else{
            return a1[1] < a2[1];
        }
    }

    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        int n = people.size();
        vector<vector<int>> ans;
        sort(people.begin(), people.end(), cmp);
        if(int i = 0; i < n; i++){
            int pos = people[i][1];
            ans.insert(ans.begin() + pos, people[i]);
        }

        return ans;
    }
};