鱼C论坛

 找回密码
 立即注册
查看: 2665|回复: 0

[学习笔记] Leetcode 498. Diagonal Traverse

[复制链接]
发表于 2020-8-2 07:19:46 | 显示全部楼层 |阅读模式

马上注册,结交更多好友,享用更多功能^_^

您需要 登录 才可以下载或查看,没有账号?立即注册

x
Given a matrix of M x N elements (M rows, N columns), return all elements of the matrix in diagonal order as shown in the below image.

 

Example:

Input:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]

Output:  [1,2,4,7,5,3,6,8,9]

Explanation:

 


Screenshot from 2020-08-01 19-15-38.png




Note:

The total number of elements of the given matrix will not exceed 10,000.
class Solution:
    def findDiagonalOrder(self, matrix: List[List[int]]) -> List[int]:
        result = []
        if matrix == None or len(matrix) == 0 or len(matrix[0]) == 0:
            return result
        m = len(matrix)
        n = len(matrix[0])
        hashmap = collections.defaultdict(list)
        for i in range(m):
            for j in range(n):
                hashmap[i + j].append(matrix[i][j])
        for k in sorted(hashmap.keys()):
            temp = hashmap[k]
            if k % 2 == 0:
                temp = temp[::-1]
            result += temp
        return result

本帖被以下淘专辑推荐:

想知道小甲鱼最近在做啥?请访问 -> ilovefishc.com
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2024-11-22 17:56

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表