04.二维数组中的查找
题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解题思路:
如果用暴力法遍历matrix,时间复杂度为O(m*n),通过引入标志数,可以降低时间复杂度。
-
标志数:左下角和右上角有特殊性,使用其作为标志数
左下角元素:所在列最大,所在行最小
右上角元素:所在行最大,所在列最小 -
算法流程:在每轮对比时消去一行(列),以降低时间复杂度
1.从矩阵 matrix 左下角元素(i,j)遍历,和target对比
matrix[ i ][ j ] > target : 行索引向上移动一格(i --)
matrix[ i ][ j ] < target : 列索引向右移动一格(j++)
matrix[ i ][ j ] == target :返回True
2.越界和矩阵内无目标值,返回False -
复杂度分析:
时间复杂度:O(M+N)
空间复杂度:O(1)
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]):
if matrix[i][j] > target : i -= 1
elif matrix[i][j] < target : j += 1
else : return True
return False