1. 问题描述:
给你一个列表 nums ,里面每一个元素都是一个整数列表。请你依照下面各图的规则,按顺序返回 nums 中对角线上的整数。
示例 1:
输入:nums = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,4,2,7,5,3,8,6,9]
示例 2:
输入:nums = [[1,2,3,4,5],[6,7],[8],[9,10,11],[12,13,14,15,16]]
输出:[1,6,2,8,7,3,9,4,12,10,5,13,11,14,15,16]
示例 3:
输入:nums = [[1,2,3],[4],[5,6,7],[8],[9,10,11]]
输出:[1,4,2,5,3,8,6,9,7,10,11]
示例 4:
输入:nums = [[1,2,3,4,5,6]]
输出:[1,2,3,4,5,6]
提示:
1 <= nums.length <= 10^5
1 <= nums[i].length <= 10^5
1 <= nums[i][j] <= 10^9
nums 中最多有 10^5 个数字。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/diagonal-traverse-ii
2. 思路分析:
① 题目不难以理解,从题目中可以看到同一对角线上的对应的横坐标与纵坐标的和是相等的,那么就可以利用这个特点将同一对角线上的元素放到map中即可,因为一开始放置的时候是从上往下放置的,而我们要求的是从下往上的顺序,所以在遍历map中的数据的时候需要倒着来进行遍历
② 需要注意应该使用HashMap有点数据不能够通过,在评论区中看到可以使用LinkedHashMap来保存插入元素的顺序可以保证在遍历map的时候是按照插入的顺序进行访问的
3. 代码如下:
class Solution {
public int[] findDiagonalOrder(List<List<Integer>> nums) {
Map<Integer, List<Integer>> map = new LinkedHashMap<>();
int len = 0;
for (int i = 0; i < nums.size(); ++i){
len += nums.get(i).size();
for (int j = 0; j < nums.get(i).size(); ++j){
List<Integer> cur = map.getOrDefault(i + j, new ArrayList<>());
cur.add(nums.get(i).get(j));
map.put(i + j, cur);
}
}
int []res = new int[len];
int pos = 0;
for (int key : map.keySet()){
List<Integer> cur = map.get(key);
for (int i = cur.size() - 1; i >= 0; --i){
res[pos++] = cur.get(i);
}
}
return res;
}
}