题目
如果一个矩阵的每一方向由左上到右下的对角线上具有相同元素,那么这个矩阵是托普利茨矩阵。
给定一个 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:
def isToeplitzMatrix(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: bool
"""
m = len(matrix)
n = len(matrix[0])
i = 0
for j in range(n-2,-1,-1):
x = matrix[i][j]
k = i+1
p = j+1
while k<m and p<n:
if matrix[k][p] == x:
k = k+1
p = p+1
else:
return False
j = 0
for i in range(1,m):
x = matrix[i][j]
k = i+1
p = j+1
while k<m and p<n:
if matrix[k][p] == x:
k = k+1
p = p+1
else:
return False
return True
学习他人代码
除了第一行和第一列元素之外,判断其余的每个元素是否等于左上角的元素,即行数减1,列数减1。
class Solution:
def isToeplitzMatrix(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: bool
"""
row = len(matrix)
col = len(matrix[0])
for i in range(row):
for j in range(col):
if i-1 >= 0 and j-1 >= 0:
if matrix[i][j] != matrix[i-1][j-1]:
return False
return True