leetcode刷题. 406. 根据身高重建队列. 贪心算法练习

最新推荐文章于 2022-04-21 22:15:17 发布

原创最新推荐文章于 2022-04-21 22:15:17 发布 · 382 阅读

0 ·

CC 4.0 BY-SA版权

【算法】同时被 2 个专栏收录

114 篇文章

订阅专栏

【leetcode刷题】

95 篇文章

订阅专栏

本文介绍了一种解决特定队列重建问题的算法。该算法针对一群由身高和相对位置描述的人，通过排序和插入操作，精确地重建了原始队列。文章详细解释了算法的工作原理，并提供了完整的代码实现。

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

注意：
总人数少于1100人。

vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
    
        vector<vector<int>> output;
        std::sort(people.begin(), people.end(), greater<vector<int>>());

        int g = 0;
        while (g < people.size()) {

                int r = g;
                while (r + 1 < people.size() && people[r][0] == people[r + 1][0])
                        r++;

                for (int i = r; i >= g; i--) {

                        if (people[i][1] < output.size()) {

                                output.insert(output.begin() + people[i][1], people[i]);
                                continue;
                        }
                        output.push_back(people[i]);
                }
                g = r + 1;
        }
        return output;
}