leetcode 406. 根据身高重建队列

笨比master

于 2021-04-04 17:33:26 发布

阅读量154

点赞数

分类专栏： new leetcode 文章标签：队列

本文链接：https://blog.youkuaiyun.com/m0_37454852/article/details/115431040

版权

new leetcode 专栏收录该内容

84 篇文章

订阅专栏

这篇博客介绍了一个C++实现的算法，用于解决根据人物的身高和前面挡着的人数来重建队列的问题。文章中提供的解决方案首先对输入数据进行排序，然后通过迭代插入每个元素到正确的位置，确保了队列的正确构造。该算法涉及到排序、列表操作和条件判断，是数据结构和算法领域的一个经典应用。

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

406. 根据身高重建队列

在这里插入图片描述

class Solution {
public:
    static bool cmp(const vector<int>& A, const vector<int>& B) {
        if(A[1] != B[1]) return A[1] < B[1];
        return A[0] < B[0];
    }
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(), people.end(), cmp);
        list<vector<int> > ans;
        for(auto p : people) {
            if(ans.empty() || p[1] == 0) ans.push_back(p);
            else {
                int cnt = 0;
                auto it = ans.begin();
                while(it != ans.end()) {
                    if((*it)[0] >= p[0]) {
                        cnt++;
                        if(cnt > p[1]) break;
                    }
                    it++;
                }
                ans.insert(it, p);
            }
        }
        return vector<vector<int>>(ans.begin(), ans.end());
    }
};