python-leetcode-766. 托普利茨矩阵

766. 托普利茨矩阵 - 力扣（LeetCode）

你可以使用 Python 来检查一个矩阵是否是托普利茨矩阵。具体方法如下：

方法

• 遍历矩阵的每个元素（从 matrix[0][0] 开始）。

• 对于每个元素 matrix[i][j]，检查它是否等于 matrix[i+1][j+1]（如果 i+1 和 j+1 在矩阵范围内）。

• 如果所有对角线上的元素都相等，则返回 True，否则返回 False。

代码

def isToeplitzMatrix(matrix):
    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]:
                return False
    return True

# 测试
matrix1 = [
    [1, 2, 3, 4],
    [5, 1, 2, 3],
    [9, 5, 1, 2]
]
matrix2 = [
    [1, 2],
    [2, 2]
]

print(isToeplitzMatrix(matrix1))  # 输出 True
print(isToeplitzMatrix(matrix2))  # 输出 False

复杂度分析

• 时间复杂度：O(m×n)，因为我们遍历整个矩阵一次。

• 空间复杂度：O(1)，没有使用额外的存储空间。

这个方法高效且简洁，适用于大多数情况。你可以试试看，有问题随时问我！ 😊