跳转至

LeetCode: 766. 托普利茨矩阵

1、题目描述

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

给定一个 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. 如果矩阵太大以至于只能一次将部分行加载到内存中,该怎么办?

2、解题思路

class Solution:
    def isToeplitzMatrix(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: bool
        """
        prev_sum = sum(matrix[0])
        current_sum = 0
        prev = matrix[0][len(matrix[0]) - 1]
        first = True
        for i in matrix:
            if first:
                first = False;
                continue
            current_sum = sum(i)
            if prev_sum - prev + i[0] != current_sum:
                return False
            prev_sum = current_sum
            prev = i[len(i) - 1]

        return True