【LeetCode笔记】406. 根据身高重建队列（Java、偏数学）

最新推荐文章于 2024-12-04 14:33:36 发布

原创最新推荐文章于 2024-12-04 14:33:36 发布

· 241 阅读

0 ·

版权

文章标签：

#数据结构 #算法 #leetcode #java #偏数学

LeetCode要每天都刷噢专栏收录该内容

167 篇文章

订阅专栏

本文介绍了如何根据身高和人数对队列进行重构。通过先按身高降序、人数升序排序，然后按照前面的人数插入到答案列表中，实现队列的重构。代码简洁，仅需六行即可完成。在二刷时，同样遵循这个思路，实现了相同的功能。

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

题目描述

老阅读题了= =，第一遍看题的时候一脸懵逼
但其实代码就6行（嘿嘿）

思路 && 代码

先按照身高降序排，再按照人数升序排（套路题）

class Solution {
    public int[][] reconstructQueue(int[][] people) {
        // 身高降序，人数升序
        Arrays.sort(people, (p1, p2) -> ((p1[0] == p2[0] ? p1[1] - p2[1] : p2[0] - p1[0])));
        List<int[]> ans = new ArrayList<>();
        // 排序后的数组，按照【前面的人数个数】进行插入即可
        for(int[] temp : people) {
            ans.add(temp[1], temp);
        }
        return ans.toArray(new int[ans.size()][]);
    }
}
// 主要原理：考虑后效性，较大元素的 ki 值，实际上和较小值无关。

无注释版

class Solution {
    public int[][] reconstructQueue(int[][] people) {
        Arrays.sort(people, (p1, p2) -> ((p1[0] == p2[0] ? p1[1] - p2[1] : p2[0] - p1[0])));
        List<int[]> ans = new ArrayList<>();
        for(int[] temp : people) {
            ans.add(temp[1], temp);
        }
        return ans.toArray(new int[ans.size()][]);
    }
}

二刷

这类题写着还是舒服，代码少= =

class Solution {
    public int[][] reconstructQueue(int[][] people) {
        // 身高逆序，人数正序
        Arrays.sort(people, (int[] a, int[] b) -> (a[0] == b[0] ? a[1] - b[1] : b[0] - a[0]));
        List<int[]> ans = new ArrayList<>();
        // 高的先插；按照前面的人个数，进行插入（往后，更矮 or 人数更多的人，不会改变之前的数组的人数正确性，无后效性）
        for(int[] temp : people) {
            ans.add(temp[1], temp);
        }
        return ans.toArray(new int[ans.size()][]);
    }
}