766. 托普利茨矩阵

最新推荐文章于 2021-09-27 19:54:09 发布

「已注销」

最新推荐文章于 2021-09-27 19:54:09 发布

阅读量149

点赞数

文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/qq_27336027/article/details/113939434

版权

托普利茨矩阵矩阵算法 Python 对角线元素比较

关键词由优快云通过智能技术生成

托普利茨矩阵

给你一个 m x n 的矩阵 matrix 。如果这个矩阵是托普利茨矩阵，返回 true ；否则，返回 false 。

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

示例 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]” 上的元素不同。

提示：

m == matrix.length
n == matrix[i].length
1 <= m, n <= 20
0 <= matrix[i][j] <= 99

解析：

col_len和row_len两个元素分别代表矩阵的行数和列数，当行为1或列为1时，返回true。
写一个for循环，依次判断第i行的第i个元素到最后一个元素与下一行的第i+1行的i+1个元素到最后一个元素是否相等。若相等则为true，否则为false

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