代码思路:
“根节点” 对应的是矩阵的 “左下角” 和 “右上角” 元素,以 matrix 中的左下角元素为标志数 flag ,则有:
若 flag > target ,则 target 一定在 flag 所在行的上方 ,即 flag 所在行可被消去,i–
若 flag < target ,则 target 一定在 flag 所在 列的右方 ,即 flag 所在列可被消去,j++
算法流程:
从矩阵 matrix 左下角元素(索引设为 (i, j) )开始遍历,并与目标值对比:
当 matrix[i][j] > target 时,执行 i-- ,即消去第 i 行元素。
当 matrix[i][j] < target 时,执行 j++ ,即消去第 j 列元素。
当 matrix[i][j] = target 时,返回 true ,代表找到目标值。
class Solution:
def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
i,j = len(matrix)-1,0
while i >= 0 and j < len(matrix[0]):
if matrix[i][j] > target: i-=1
elif matrix[i][j] < target: j+=1
else:return True
return False