蓝色风格网站网站案例上海
题目来源：力扣
此解法是在力扣官网上找到的
原作者：Storm
链接：1424. 对角线遍历 II - 力扣（LeetCode）
加上了自己的理解与分析便于理解代码
// 对角线遍历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. 定义类和方法
class Solution {public int[] findDiagonalOrder(List<List<Integer>> nums) {-
定义了一个名为
Solution的类。 -
定义了一个公共方法
findDiagonalOrder，它接收一个嵌套列表nums作为参数，并返回一个整数数组。
2. 创建存储元素信息的列表
List<int[]> list = new ArrayList<int[]>();int rows = nums.size();-
创建了一个
List<int[]>类型的list，用于存储每个元素的信息。每个元素信息用一个长度为 3 的整数数组表示，分别存储i + j、j和元素值。（用于后面排序） -
rows变量存储了嵌套列表的行数。
3. 遍历嵌套列表
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. 对元素信息列表进行排序（重点排序）
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. 创建结果数组并存储排序后的元素
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. 返回结果数组
return order;-
返回存储排序后元素的数组
order。
复杂度分析
-
时间复杂度：O(Nlog**N)，其中 N 是嵌套列表中元素的总数。主要时间开销在于对
list进行排序。 -
空间复杂度：O(N)，主要空间开销在于存储元素信息的
list和结果数组order。
有更好的解法，欢迎评论或私信