力扣刷题记录#数组#简单#766托普利茨矩阵

题目描述

如果一个矩阵的每一方向由左上到右下的对角线上具有相同元素，则这个矩阵是托普利茨矩阵。
给定一个 M x N 的矩阵，当且仅当它是托普利茨矩阵时返回 True。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/toeplitz-matrix
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例

输入:
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。

输入:
matrix = [
[1,2],
[2,2]
]
输出: False
解释: 对角线"[1, 2]"上的元素不同。

解答

遍历矩阵，判断其是否与其左上角元素一致

class Solution(object):
    def isToeplitzMatrix(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: bool
        """
        
        row = len(matrix)   # 矩阵行数
        col = len(matrix[0])   # 矩阵列数
        
        # 从[1,1]开始遍历矩阵，判断是否与其左上元素相等
        for r in range(1,row):
            for c in range(1,col):
                if matrix[r][c] != matrix[r-1][c-1]:
                    return False
                
        return True