766. 托普利茨矩阵（简单）- LeetCode

最新推荐文章于 2023-03-07 16:46:15 发布

不学无术的小镇做题家

最新推荐文章于 2023-03-07 16:46:15 发布

阅读量204

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法数据结构 python

本文链接：https://blog.youkuaiyun.com/Karty9/article/details/107291605

版权

LeetCode 专栏收录该内容

52 篇文章

订阅专栏

题目描述

在这里插入图片描述

自己解法

对于一个 $M \times N$ 矩阵，遍历前 $M - 1$ 行的前 $N - 1$ 个元素，假设下标为 $(i, j)$ ，若该元素与下标为 $(i + 1, j + 1)$ 的元素相等，则满足题目条件，若不相等，则不满足。时间复杂度 $O (M \times N)$ ，空间复杂度 $O (1)$ 。

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

在这里插入图片描述

题解区

除了自己解法以外，还有一种解法是检查对角线元素：

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

与自己解法类似的简易代码是：

class Solution(object):
    def isToeplitzMatrix(self, matrix):
        return all(r == 0 or c == 0 or matrix[r-1][c-1] == val
                   for r, row in enumerate(matrix)
                   for c, val in enumerate(row))