在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路:每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。如下图所示,例如18这个数,在其所在列的上方的数都比18小,所在行的右侧的数都比18大,根据这个关系,我们可以将这个矩阵左下角的数当作标志数,可以得到:
1)当目标值>标志数时,删除所在列
2)当目标值<标志数时,删除所在行
3)当目标值=标志数时,返回true,代表找到了目标值
class Solution(object):
def findNumberIn2DArray(self, matrix, target):
"""
:type matrix: List[List[int]]
:type target: int
:rtype: 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