LeetCode-Python-766. 托普利茨矩阵

给你一个 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

进阶：

• 如果矩阵存储在磁盘上，并且内存有限，以至于一次最多只能将矩阵的一行加载到内存中，该怎么办？
• 如果矩阵太大，以至于一次只能将不完整的一行加载到内存中，该怎么办？

思路：

遍历左上到右下的对角线， 模拟即可。注意在对角线上移动是 [x, y] -> [x + 1, y + 1]。

时间复杂度：O(MN)

空间复杂度：O（M + N），可以优化到O（1）

class Solution:
    def isToeplitzMatrix(self, matrix: List[List[int]]) -> bool:
        if not matrix or not matrix[0]:
            return false
        m, n = len(matrix), len(matrix[0])
        cors = [(0, j) for j in range(n)] + [(i, 0) for i in range(1, m)]
        for i, j in cors:
            val = matrix[i][j]
            while 0 <= i < m and 0 <= j < n:
                if matrix[i][j] != val:
                    return False
                i += 1
                j += 1

        return True            

进阶：

我们可以在内存里存储一个哈希表，它有(m + n - 1) 个key，value是对应的值。

在一条左上到右下的对角线上的点，有一个特征：它们 (x - y) 是固定的。比如说 (0, 0), (1, 1) 都有 x - y == 0。

用这种方法，我们可以知道每个点在哪条对角线上，从而可以比较当前值和应有值。