给你一个 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。
用这种方法,我们可以知道每个点在哪条对角线上,从而可以比较当前值和应有值。