[剑指offer 04] 二维数组的查找

剑指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