[leecode]Python:766 Toeplitz Matrix托普利茨矩阵

如果一个矩阵的每一方向由左上到右下的对角线上具有相同元素,那么这个矩阵是托普利茨矩阵。

给定一个 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]"上的元素不同。

说明:

  1. matrix 是一个包含整数的二维数组。
  2. matrix 的行数和列数均在 [1, 20]范围内。
  3. matrix[i][j] 包含的整数在 [0, 99]范围内。
    代码如下
class Solution:
    def isToeplitzMatrix(self, A: List[List[int]]) -> bool:
        for i in range(len(A)-1):
            for y in range(len(A[0])-1):
                if A[i][y] != A[i+1][y+1]:
                    return False
        return True

非常简单的思路,直接将数组的每个元素跟它的下一个做对比,相同则为true,而边角的数值则不用做对比,因为只有一个

你可能感兴趣的:(leecode)