Leetcode 406. 根据身高重建队列

Z-Pilgrim

于 2023-06-18 12:10:37 发布

阅读量170

点赞数

文章标签： c++ 算法开发语言

本文链接：https://blog.youkuaiyun.com/u011026968/article/details/131270077

版权

该代码实现了一个解决方案，用于根据身高和前面可见人数对人进行排序。首先，使用自定义的cmp函数按身高降序，然后按前面可见人数升序对人进行排序。接着，遍历排序后的队列，将每个人插入到正确的位置，确保满足题目要求。

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

力扣

class Solution {
public:
    static bool cmp(vector<int>&a, vector<int>&b) {
        return a[0] == b[0] ? a[1] < b[1] : a[0] > b[0];
    }

    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(), people.end(), cmp);
        // [[7,0],[7,1],[6,1],[5,0],[5,2],[4,4]]
        // return people;
        for(int i = 1; i < people.size(); i++) {
                int pos = 0, cnt = 0;
                for(;pos < i; pos++ ) {
                    if(cnt == people[i][1]) {
                        vector<int> tmp = people[i];
                        for(int j = i; j > pos; j--) {
                            people[j] = people[j-1];
                        } 
                        people[pos] = tmp;
                        break;
                    }
                    if(people[pos][0] >= people[i][0]) {
                        cnt ++;
                    }
                    
                }
        }
        return people;
    }
};