406. Queue Reconstruction by Height

最新推荐文章于 2023-11-18 01:25:37 发布

原创最新推荐文章于 2023-11-18 01:25:37 发布 · 149 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法分析与设计

本文介绍了一种队列重建算法，该算法通过排序和插入操作，实现根据每个人的高度及身前高个子人数来重建队列。具体步骤包括：按身高降序、身高相同时按k值升序进行排序，并根据k值将元素插入结果队列。

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

题目：假设现在有一队列，队列中的人随机站立。每个人都被一个数对（h，k）所描述，h表示这个人的身高，k表示站在这个人前面身高比他高的人数。要求我们重新排列顺序，使得每个人的站位都符合数对的描述。例子如下：

Input:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]

Output:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]

思路是按照身高从高到矮排序，如果身高相同，则按照k值从小到大排序，排序的代码如下：

static bool compare(pair<int, int> a, pair<int, int> b) {
        return (a.first == b.first ? a.second < b.second : a.first > b.first);
    }

这样排完后，再直接从头开始遍历，根据k值确定在队伍中的插入位置：

vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
        vector<pair<int, int>> result;
        sort(people.begin(), people.end(), compare);
        for (int i = 0; i < people.size(); i++) {
            result.insert(result.begin() + people[i].second, people[i]);
        }
        return result;
    }