【LeetCode】【算法】406. 根据身高重建队列

LeetCode 406. 根据身高重建队列

题目描述

假设有打乱顺序的一群人站成一个队列,数组 people 表示队列中一些人的属性(不一定按顺序)。每个 people[i] = [h_i, k_i] 表示第 i 个人的身高为 h_i ,前面 正好 有 k_i 个身高大于或等于 h_i 的人。

请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ,其中 queue[j] = [h_j, k_j] 是队列中第 j 个人的属性(queue[0] 是排在队列前面的人)。

思路

参考的题解【先排序,再插队】动画演示算法过程,有点小套路

  1. 先做排序,排序思路是h_i降序,k_i升序(这么排序是为了保证正确性,如果k_i是降序排列的话,假设有(5,5),先插入到队列之后,又插入(5,1),但此时(5,5)前面排队的人多了一个,就导致不符合我们目标要求)
  2. 排序完成后,将结果往list里放,放的规则是:
    I. 如果列表当前长度<k_i的话,直接将这个人插入到列表末尾
    II. 否则将这个人插入到列表的k_i处

代码

java 复制代码
class Solution {
    public int[][] reconstructQueue(int[][] people) {
        // 第一个元素降序,第二个元素升序
        Arrays.sort(people, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                if (o1[0] != o2[0]){
                    // 第一个元素不相等,降序排列
                    return o2[0] - o1[0];
                } else {
                    // 第二个元素升序排列
                    return o1[1] - o2[1];
                }
            }
        });
        // 排序过后,往结果里放
        List<int[]> list = new LinkedList<>();
        for (int i = 0; i < people.length; i++) {
            if (list.size() < people[i][1]){
                list.add(people[i]);
            } else {
                list.add(people[i][1], people[i]);
            }
        }
        return list.toArray(new int[list.size()][]);
    }
}
相关推荐
浊酒南街1 分钟前
决策树(理论知识1)
算法·决策树·机器学习
就爱学编程9 分钟前
重生之我在异世界学编程之C语言小项目:通讯录
c语言·开发语言·数据结构·算法
学术头条13 分钟前
清华、智谱团队:探索 RLHF 的 scaling laws
人工智能·深度学习·算法·机器学习·语言模型·计算语言学
Schwertlilien1 小时前
图像处理-Ch4-频率域处理
算法
IT猿手1 小时前
最新高性能多目标优化算法:多目标麋鹿优化算法(MOEHO)求解TP1-TP10及工程应用---盘式制动器设计,提供完整MATLAB代码
开发语言·深度学习·算法·机器学习·matlab·多目标算法
__lost1 小时前
MATLAB直接推导函数的导函数和积分形式(具体方法和用例)
数学·算法·matlab·微积分·高等数学
thesky1234561 小时前
活着就好20241224
学习·算法
ALISHENGYA1 小时前
全国青少年信息学奥林匹克竞赛(信奥赛)备考实战之分支结构(实战项目二)
数据结构·c++·算法
guogaocai1231 小时前
连续自成核退火热分级(SSA)技术表征共聚聚丙烯(PP)分子链结构
算法
DARLING Zero two♡2 小时前
【优选算法】Pointer-Slice:双指针的算法切片(下)
java·数据结构·c++·算法·leetcode