1424.对角线遍历

题目来源:力扣

此解法是在力扣官网上找到的

原作者:Storm

链接:1424. 对角线遍历 II - 力扣(LeetCode)

加上了自己的理解与分析便于理解代码

java 复制代码
 //    对角线遍历2
 //    给你一个列表 nums ,里面每一个元素都是一个整数列表。
 //    请你依照下面各图的规则,按顺序返回 nums 中对角线上的整数。
 ​
 //    每一个子列表代表一行,不同行的长度可能不同
         public static int[] findDiagonalOrder(List<List<Integer>> nums) {
         List<int[]> list = new ArrayList<int[]>();
         int rows = nums.size();
         for (int i = 0; i < rows; i++) {
             List<Integer> rowList = nums.get(i);
             int cols = rowList.size();
             for (int j = 0; j < cols; j++) {
                 int num = rowList.get(j);
                 list.add(new int[]{i + j, j, num});
             }
         }
         Collections.sort(list, (a, b) -> {
             if (a[0] != b[0]) {
                 return a[0] - b[0];
             } else {
                 return a[1] - b[1];
             }
         });
         int size = list.size();
         int[] order = new int[size];
         for (int i = 0; i < size; i++) {
             order[i] = list.get(i)[2];
         }
         return order;
     }

代码详细解释

1. 定义类和方法
java 复制代码
 class Solution {
     public int[] findDiagonalOrder(List<List<Integer>> nums) {
  • 定义了一个名为 Solution 的类。

  • 定义了一个公共方法 findDiagonalOrder,它接收一个嵌套列表 nums 作为参数,并返回一个整数数组。

2. 创建存储元素信息的列表
java 复制代码
List<int[]> list = new ArrayList<int[]>();
 int rows = nums.size();
  • 创建了一个 List<int[]> 类型的 list,用于存储每个元素的信息。每个元素信息用一个长度为 3 的整数数组表示,分别存储 i + jj 和元素值。(用于后面排序)

  • rows 变量存储了嵌套列表的行数。

3. 遍历嵌套列表
java 复制代码
 for (int i = 0; i < rows; i++) {
     List<Integer> rowList = nums.get(i);
     int cols = rowList.size();
     for (int j = 0; j < cols; j++) {
         int num = rowList.get(j);
         list.add(new int[]{i + j, j, num});
     }
 }
  • 外层循环遍历每一行,获取当前行的子列表 rowList 以及当前行的列数 cols

  • 内层循环遍历当前行的每一列,获取当前元素 num,并将 [i + j, j, num] 存储到 list 中。

4. 对元素信息列表进行排序(重点排序)
java 复制代码
Collections.sort(list, (a, b) -> {
     if (a[0] != b[0]) {
         return a[0] - b[0];
     } else {
         return a[1] - b[1];
     }
 });
  • 使用 Collections.sort 方法对 list 进行排序。

  • 排序规则是:先比较元素的 i + j 值(即 a[0]b[0]),若不相等,则按照 i + j 从小到大排序;若相等,则比较列索引 j(即 a[1]b[1]),按照列索引从小到大排序。

5. 创建结果数组并存储排序后的元素
java 复制代码
 int size = list.size();
 int[] order = new int[size];
 for (int i = 0; i < size; i++) {
     order[i] = list.get(i)[2];
 }
  • size 存储了 list 的大小,即元素的总数。

  • 创建一个长度为 size 的整数数组 order

  • 遍历 list,将每个元素信息数组中的第三个元素(即元素值)存储到 order 数组中。

6. 返回结果数组
java 复制代码
 return order;
  • 返回存储排序后元素的数组 order

复杂度分析

  • 时间复杂度O (Nlog**N ),其中 N 是嵌套列表中元素的总数。主要时间开销在于对 list 进行排序。

  • 空间复杂度O (N ),主要空间开销在于存储元素信息的 list 和结果数组 order

有更好的解法,欢迎评论或私信

相关推荐
知来者逆3 分钟前
计算机视觉——速度与精度的完美结合的实时目标检测算法RF-DETR详解
图像处理·人工智能·深度学习·算法·目标检测·计算机视觉·rf-detr
阿让啊7 分钟前
C语言中操作字节的某一位
c语言·开发语言·数据结构·单片机·算法
এ᭄画画的北北8 分钟前
力扣-160.相交链表
算法·leetcode·链表
爱研究的小陈1 小时前
Day 3:数学基础回顾——线性代数与概率论在AI中的核心作用
算法
渭雨轻尘_学习计算机ing1 小时前
二叉树的最大宽度计算
算法·面试
BB_CC_DD2 小时前
四. 以Annoy算法建树的方式聚类清洗图像数据集,一次建树,无限次聚类搜索,提升聚类搜索效率。(附完整代码)
深度学习·算法·聚类
梁下轻语的秋缘3 小时前
每日c/c++题 备战蓝桥杯 ([洛谷 P1226] 快速幂求模题解)
c++·算法·蓝桥杯
CODE_RabbitV3 小时前
【深度强化学习 DRL 快速实践】逆向强化学习算法 (IRL)
算法
mit6.8244 小时前
[贪心_7] 最优除法 | 跳跃游戏 II | 加油站
数据结构·算法·leetcode
keep intensify4 小时前
通讯录完善版本(详细讲解+源码)
c语言·开发语言·数据结构·算法