- 对角线遍历 II
给你一个列表 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 个数字。
思路:这道题目思路就很明显,对角线上的元素下标之和都一样
比如8是(0,2), 7是(1,1),3是(2, 0)
所以我就可以维护一个set保存当前下标之和 和 对应的数字皆可
对角线元素的下标之和是相等的,所以可以放在同一个坐标上面保存
m = len(nums)
n = max([len(num_) for num_ in nums])
my_list = [[] for _ in range(m+n-1)]
for num_arr in nums: