Leetcode 406. Queue Reconstruction by Height

最新推荐文章于 2025-05-19 11:54:59 发布

原创最新推荐文章于 2025-05-19 11:54:59 发布 · 257 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

刷题专栏收录该内容

24 篇文章

订阅专栏

本文介绍了一种使用贪心策略的算法来解决特定队列重构问题。该算法通过将人群按身高降序排序，并在遇到相同身高时根据前面较高人数升序排序的方法，有效地解决了问题。最终实现了对输入数据的有效重组。

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

-题目-
Suppose you have a random list of people standing in a queue. Each person is described by a pair of integers (h, k), where h is the height of the person and k is the number of people in front of this person who have a height greater than or equal to h. Write an algorithm to reconstruct the queue.

Note:
The number of people is less than 1,100.
Example
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]]

-思路-
这是一道贪心算法的题目，每次加入一个人，使得当前队伍的重排满足条件，这样最终加入的人也能使整个队伍满足重排的条件。为了使贪心算法有最大效益，需要对输入的原队伍进行排序，由题意可知，矮的人对高的人的排序不产生影响，因此先从高的人开始重排。然后将身高矮的人不断加进前面的人已产生好的排列。需要注意的是，当身高相同时，需要再根据前面的更高的人数按照从小到大的顺序进行排序，因为前面更高人数比较小的人会对前面更高的人数比较大的人产生影响。（因为它可能插到前面更高人数比较大的人前面，从而导致错误）

-代码-

class Solution {
public:
    static bool cmp1(pair<int,int> &a, pair<int,int> &b) {
        return a.first > b.first || ((a.first == b.first) && (a.second < b.second)); 
    }
    vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
        vector<pair<int, int> > newQueue; 
        //按身高和前面比他高的人排序
        sort(people.begin(), people.end(), cmp1); 
        for(int i = 0; i < people.size() ; i++) {
            cout << people[i].first << " " << people[i].second << endl; 
        }

        if(people.size() == 0) return newQueue; 

        newQueue.push_back(people[0]); 
        vector<pair<int, int>>::iterator i, j; 
        for(i = people.begin()+1; i != people.end(); i++) {
            int s = i -> second; 
            bool isin = 0; 
            for(j = newQueue.begin(); j != newQueue.end(); j++) {
                if(!s) {
                    newQueue.insert(j, *i); 
                    isin = 1; 
                    break; 
                }
                else if(j->first >= i->first) s--; 
            }
            if(!isin) newQueue.insert(newQueue.end(), *i); 
        }
        return newQueue; 
    }
};