剑指04 二维数组的查找
**题目描述:**在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
方法一【错】把输一个个add到set()中,检查是否在其中
由于矩阵中可能相同的两个数,这种方法不可行
class Solution:
def findNumberIn2DArray(self, matrix: List[List[int]], target: int) -> bool:
dic = set()
dic.add(target)
i,j = 0,0
while i < len(matrix):
for num in matrix[i]:
if num in dic:
return True
dic.add(num)
i +=1
return False
方法二 利用行列都是递增的特点
设左下角的数为flag,那么flag为所在列的最大数、所在行的最小数;比较左下角flag与target的大小,若flag>target,则去掉当前列,否则去掉当前行
class Solution:
def findNumberIn2DArray(self, matrix: List[List[int]], target: int) -> bool:
i, j = len(matrix) - 1, 0
while i >= 0 and j < len(matrix[0]): #循环条件的不同;i,j不断加减就好
if matrix[i][j] > target: i -= 1
elif matrix[i][j] < target: j += 1
else: return True
return False