题:在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。
- 从二维数组的右上角开始查找。如果当前元素大于目标值,则移到左边一列。如果当前元素小于目标值,则移到下边一行。如果当前元素等于目标值,则返回
true
。
class Solution:
def findNumberIn2DArray(self, matrix: List[List[int]], target: int) -> bool:
if not matrix:
return False
# 行len(matrix) 列len(matrix[0])
i,j=0,len(matrix[0])-1 #从右上角查找
while i<len(matrix) and j>=0:
if matrix[i][j] <target:
i+=1 #往下一行走
elif matrix[i][j] >target:
j-=1 #往左一列走
elif matrix[i][j]==target:
return True
return False