Leetcode406. 根据身高重建队列 -代码随想录-春招冲刺

meeiuliuus

已于 2024-06-13 23:37:45 修改

阅读量412

点赞数 6

CC 4.0 BY-SA版权

分类专栏： # leetcode ---medium 文章标签：算法

于 2024-02-15 15:28:53 首次发布

本文链接：https://blog.youkuaiyun.com/qq_52313711/article/details/136120752

leetcode ---medium 专栏收录该内容

174 篇文章

订阅专栏

本文介绍了两种Solution类的实现，用于解决根据身高和到达时间重构队列的问题。首先按身高降序排列，再根据到达时间升序调整位置，使用C++STL中的list实现。

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

题目：

代码(首刷看解析 2024年2月15日）：

class Solution {
static bool cmp(const vector<int>& A, const vector<int>& B) {
    if(A[0] == B[0]) return A[1] < B[1];
    return A[0] > B[0];
}
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(), people.end(), cmp);
        list<vector<int>> que;
        for (int i = 0; i < people.size(); i++) {
            int postion = people[i][1];
            list<vector<int>>::iterator it = que.begin();
            while (postion--) {
                it++;
            }
            que.insert(it, people[i]);
        }
        return vector<vector<int>>(que.begin(), que.end());
    }
};

代码（二刷看解析 2024年4月13日）

class Solution {
public:
    // 先按身高从高到低排序，相同的话矮的在前面
    static bool cmp(const vector<int>& A, const vector<int>& B) {
        if (A[0] == B[0]) return A[1] < B[1];
        return A[0] > B[0];
    }
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        // 按[i][0]的大小先排序
        sort(people.begin(), people.end(), cmp);
        // 处理[i][1]的情况
        list<vector<int>> mylist;
        for (int i = 0; i < people.size(); ++i) {
            int postion = people[i][1];
            auto it = mylist.begin();
            // position是前面应该有多少个大于自己，因为排序好的情况下前面全都大于自己
            // 所以直接从头开始往后数position个位置插入就可以了，也不会影响前面的相对顺序
            while (postion--) {
                it++;
            }
            mylist.insert(it, people[i]);
        }
        return vector<vector<int>>(mylist.begin(), mylist.end());
    }
};

代码（三刷看解析 2024年6月13日 go）

// 每个元素有h k两个参数，分开处理两个参数
func reconstructQueue(people [][]int) [][]int {
    // 先处理h，让h从小到大
    queue := people
    sort.Slice(queue, func(i, j int) bool {
        if queue[i][0] == queue[j][0] {
            return queue[i][1] < queue[j][1]
        }
        return queue[i][0] > queue[j][0]
    })
    // 再处理k,从左往右将每个元素往后移动k次
    for i,p := range queue {
        // slice[i : j] 代表从i开始（包含）到j结束（不包含）的切片
        copy(queue[p[1] + 1 : i + 1], queue[p[1] : i + 1])
        queue[p[1]] = p
    }
    return queue
}