Leetcode766. 托普利茨矩阵

题目

如果矩阵上每一条由左上到右下的对角线上的元素都相同，那么这个矩阵是 托普利茨矩阵 。
给定一个 M x N 的矩阵，当且仅当它是托普利茨矩阵时返回 True。
示例 1:
输入:
matrix = [
[1,2,3,4],
[5,1,2,3],
[9,5,1,2]
]
输出: True
解释:
在上述矩阵中, 其对角线为:
“[9]”, “[5, 5]”, “[1, 1, 1]”, “[2, 2, 2]”, “[3, 3]”, “[4]”。
各条对角线上的所有元素均相同, 因此答案是True。
示例 2:
输入:
matrix = [
[1,2],
[2,2]
]
输出: False
解释:
对角线"[1, 2]"上的元素不同。

思路：对角线上的两个数的坐标满足r1-c1=r2-c2;创建一个字典保存每个对角线上的值；

class Solution:
    def isToeplitzMatrix(self, matrix: List[List[int]]) -> bool:
        diagonal={}
        for r, row in enumerate(matrix):
            for c, val in enumerate(row):
                if r-c not in diagonal:
                    diagonal[r-c]=val
                elif diagonal[r-c]!=val:
                    return False
        return True