leetcode 766. 托普利茨矩阵(python)

题目链接

题目描述：

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

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

matrix 是一个包含整数的二维数组。
matrix 的行数和列数均在 [1, 20]范围内。
matrix[i][j] 包含的整数在 [0, 99]范围内

解题思路：
遍历数组，判断每一个元素和它右下方元素（行数和列数均加一）是否相等，注意每一行最后一个元素以及最后一行元素无需比较（已经比较过了），所以遍历时，行数和列数都要减1.

class Solution:
    def isToeplitzMatrix(self, matrix: List[List[int]]) -> bool:
        for i in range(len(matrix)-1):
            for j in range( len(matrix[0])-1) :
                if matrix[i][j]!=matrix[i+1][j+1]:
                    return False
        return True

p.s.别人的做法,不用每个元素比较一次，可以进行整行比较，这样只用一重循环。

        for i in range(1,len(matrix)):
            if matrix[i][1:]!=matrix[i-1][0:-1]:
                return False
        return True