Posted on 2022-10-31 18:01
Uriel 閱讀(68)
評論(0) 編輯 收藏 引用 所屬分類:
閑來無事重切Leet Code 、
大水題
判斷一個矩陣斜對角元素是否都一樣，大水題
1 #766
2 #Runtime: 64 ms
3 #Memory Usage: 13.5 MB
4
5 class Solution(object):
6 def isToeplitzMatrix(self, matrix):
7 """
8 :type matrix: List[List[int]]
9 :rtype: bool
10 """
11 for i in range(len(matrix[0])):
12 p = matrix[0][i]
13 for j in range(1, min(len(matrix), len(matrix[0]) - i)):
14 if matrix[j][i+j] != p:
15 return False
16 for i in range(len(matrix)):
17 p = matrix[i][0]
18 for j in range(1, min(len(matrix[0]), len(matrix) - i)):
19 if matrix[i+j][j] != p:
20 return False
21 return True