766. 托普利茨矩阵

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

给定一个 M x N 的矩阵，当且仅当它是托普利茨矩阵时返回 True。

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

说明:

1.  matrix 是一个包含整数的二维数组。
2. matrix 的行数和列数均在 [1, 20]范围内。
3. matrix[i][j] 包含的整数在 [0, 99]范围内。

进阶:

1. 如果矩阵存储在磁盘上，并且磁盘内存是有限的，因此一次最多只能将一行矩阵加载到内存中，该怎么办？
2. 如果矩阵太大以至于只能一次将部分行加载到内存中，该怎么办？

class Solution:
    '''
    1 2 3 4
    5 1 2 3
    9 5 1 2
    起点移动的方向是9 -> 5 -> 1 -> 2 -> 3 -> 4
    '''
    def isToeplitzMatrix(self, matrix: 'List[List[int]]') -> 'bool':
        row = len(matrix)
        col = len(matrix[0])
        p = row - 1
        q = 0
        while p >= 0 and q < col:
            val, i, j = matrix[p][q], p, q
            while i + 1 < row and j + 1 < col:
                i += 1
                j += 1
                if matrix[i][j] != val:
                    return False
            if p > 0:
                p = p - 1
            else:
                q = q + 1
        return True
    #按正常顺序来遍历数组，对于每个遍历到的数字，都跟其右下方的数字对比，如果不相同，直接返回false即可
    def isToeplitzMatrix2(self, matrix: 'List[List[int]]') -> 'bool':
        for i in range(len(matrix)-1):
            for j in range(len(matrix[0])-1):
                if matrix[i][j] != matrix[i+1][j+1] :
                    return False
        return True
    '''
    错位按行对比
    [1, 2, 3] == [1, 2, 3] 
    [5, 1, 2] == [5, 1, 2] 
    True
    '''
    def isToeplitzMatrix3(self, matrix: 'List[List[int]]') -> 'bool':
        for i in range(1, len(matrix)):
            #print(matrix[i][1:],' ',matrix[i - 1][0:-1],'\n')
            if matrix[i][1:] != matrix[i - 1][0:-1]:
                return False
        return True

if __name__ == '__main__':
    s = Solution()
    matrix = [
        [1, 2, 3, 4],
        [5, 1, 2, 3],
        [9, 5, 1, 2]
    ]
    print(s.isToeplitzMatrix3(matrix))