406. Queue Reconstruction by Height(Medium)

原创于 2021-10-08 21:58:08 发布 · 205 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #c++ #算法 #贪心算法

第二章专栏收录该内容

9 篇文章

订阅专栏

该博客介绍了LeetCode第406题的解决方案，通过贪心算法重新构造高度优先队列。作者首先尝试按原始顺序逐个元素入队，然后调整为按高度排序后再入队，使用C++的vector和insert函数来实现这一过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

解题思路：

初步想法：按初始顺序一个一个进入队列，进去后从头开始比较，直到不满足条件的就在那个地方入队。
代码实现后发现结果不对，存在先存进去的不满足Ki条件。故可以在进入队列前按Ki条件先排序好

在这里插入图片描述

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(), people.end(), [](vector<int>& a, vector<int>& b){
            return a[1]<b[1];
        });//排序
        vector<vector<int>> ans;
        vector<vector<int>>::iterator it;
        ans.push_back(people[0]);
        for(int i = 1; i < people.size(); ++i){
            int cnt = 0;
            for(it = ans.begin(); it != ans.end(); ++it){
                vector<int> j = *it;
                if(people[i][0] <= j[0]){//比较高度
                    if(cnt+1> people[i][1]) break;//判断Ki条件
                    else{
                        cnt += 1;
                    }
                }
            }
            ans.insert(it,people[i]);
        }
        return ans;
    }
};

收获：

vector容器用insert函数加迭代器实现插入。