[C++]Leetcode406.根据身高重建队列

最新推荐文章于 2024-01-20 14:07:06 发布

原创最新推荐文章于 2024-01-20 14:07:06 发布 · 589 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#c++ #算法

这是一个关于算法实现的博客，主要讲解了如何解决LeetCode上的406题。题目要求根据一组人的身高和前面比自己高的人数，重建原来的队列顺序。解决方案是首先对输入的人员列表进行排序，优先考虑身高较高的个体，并在身高相同时，依据在队列前有身高不低于自己的人数（k值）进行排序。然后按照排序后的k值将人插入到结果队列的相应位置。提供的C++代码使用了lambda表达式进行排序，并通过迭代将人插入正确位置。

406.根据身高重建队列

题目：假设有打乱顺序的一群人站成一个队列。
每个人由一个整数对(h, k)表示，其中h是这个人的身高，k是排在这个人前面且身高大于或等于h的人数。编写一个算法来重建这个队列。
注意：总人数少于1100人。

示例：
输入: [[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
输出:[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) 
    {
        //排序：将最高的人按照 k 值升序排序，然后将它们放置到输出队列中与 k 值相等的索引位置上。
        //按降序取下一个高度，同样按 k 值对该身高的人升序排序，然后逐个插入到输出队列中
        sort(people.begin(), people.end(),[](const vector<int> &a,const vector<int> &b) //lambda表达式
        {
            if(a[0] > b[0]) return true;//先按身高h排序
            if(a[0] == b[0] && a[1] < b[1]) return true;//身高相同按k排序
            return false;
        });
        vector<vector<int>> res;
        for( auto &n : people)
        {
            //每个向量都按其人数k放在对应的位置上（对整体已经排好的向量而言）
            res.insert(res.begin()+n[1],n);
        }
        return res;
    }
};