如果一个矩阵的每一方向由左上到右下的对角线上具有相同元素,那么这个矩阵是托普利茨矩阵。
给定一个 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]"上的元素不同。
说明:
-
matrix
是一个包含整数的二维数组。 matrix
的行数和列数均在[1, 20]
范围内。matrix[i][j]
包含的整数在[0, 99]
范围内。
进阶:
- 如果矩阵存储在磁盘上,并且磁盘内存是有限的,因此一次最多只能将一行矩阵加载到内存中,该怎么办?
- 如果矩阵太大以至于只能一次将部分行加载到内存中,该怎么办?
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))