Leetcode 406. 根据身高重建队列

Leetcode 406. 根据身高重建队列

题目

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

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

思路

• 首先按照身高进行排序，从大到小
• 然后建立一个队列
• 循环排序过后的数组，优先取出身高的人，将其按照k的位置插入队列，前面插入的节点不会影响后面插入的节点，最终按照的规则完成了排序

代码

class Solution {
public:
    static bool cmp(const vector<int>& a,const vector<int>& b)
    {
        if(a[0] == b[0])
        {
            return a[1] < b[1];// 如果身高相等 那么按照k进行排序
        }
        return a[0] > b[0];
    }

    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        // 先按照身高进行排序
        sort(people.begin(),people.end(),cmp);
        vector<vector<int>> que;
        for(int i = 0; i < people.size(); i++)
        {
            // 优先按照身高高的people的k来进行插入
            // 这样的话 身高低的插入 不会被影响 因为身高的高的已经被安排好
            // 后序插入的节点不会影响前面已经插入的节点，最终按照k的规则完成了队列
            int position = people[i][1];// 取出k
            //插入
            que.insert(que.begin() + position,people[i]);
        }
        return que;
    }
};