题目:
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
leetcode原题链接
思路:
从矩阵(二维数组)的右上角开始遍历
若当前元素大于目标整数target,往下继续寻找
若当前元素小于目标整数target,往左继续寻找
若遍历完整个矩阵都没找到target,return False
代码:
时间复杂度O(m+n)
m为矩阵行数,n为矩阵列数,和代码对应
class Solution:
def findNumberIn2DArray(self, matrix: List[List[int]], target: int) -> bool:
if matrix == []:
return False
m=len(matrix)#m行
n=len(matrix[0])#n列
i,j=0,n-1
while i<m and j>=0:
if target>matrix[i][j]:
i+=1
elif target<matrix[i][j]:
j-=1
else:
return True
return False