leetcode-1424. 对角线遍历 II

最新推荐文章于 2026-05-23 23:35:41 发布

原创最新推荐文章于 2026-05-23 23:35:41 发布 · 347 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

Leetcode

该博客详细介绍了LeetCode第1424题——对角线遍历II的解题方法。通过举例说明了如何按照特定规则返回矩阵对角线上的数字，并提出了利用下标之和相同的特性来存储和遍历对角线元素的思路。

对角线遍历 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:
            for num in num_arr:
                x = nums.index(num_arr)
                y = num_arr.index(num)
                my_list[x+y].append(num)
        ret = []
        for nums in my_list:
            nums.reverse()
            for num in nums:
                ret.append(num)

        return ret



from typing import List


class Solution:
    def findDiagonalOrder(self, nums: List[List[int]]) -> List[int]:
        m = len(nums)
        n = max([len(num_) for num_ in nums])
        my_list = [[] for _ in range(m+n-1)]

        for num_arr in nums:
            for num in num_arr:
                x = nums.index(num_arr)
                # 所以这一不会出错，看下面那一版的代码，动态寻找
                y = num_arr.index(num)  # 出现问题，因为index是获取的是第一个出现的下标
                my_list[x+y].append(num)
        ret = []
        for nums in my_list:
            if not nums:
                continue
            nums.reverse()
            for num in nums:
                ret.append(num)

        return my_list

    def findDiagonalOrder_02(self, nums: List[List[int]]) -> List[int]:
        ret = []
        for i, row in enumerate(nums):
            for j, num in enumerate(row):
                if i + j >= len(ret):
                    ret.append([])
                ret[i + j].append(num)
        my_list = []
        for nums in ret:
            nums.reverse()
            for num in nums:
                my_list.append(num)
        return my_list



s = Solution()
print(s.findDiagonalOrder_02([[5,6,3,10],[9],[1,19],[9,9,2]]))
# c = [[1,2], [3,4]]
# print(c.index([3,4]))