剑指 Offer 04. 二维数组中的查找

问题描述：
在一个 n * m 的二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个高效的函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

示例:
现有矩阵 matrix 如下：

[
[1, 4, 7, 11, 15],
[2, 5, 8, 12, 19],
[3, 6, 9, 16, 22],
[10, 13, 14, 17, 24],
[18, 21, 23, 26, 30]
]
给定 target = 5，返回 true。

给定 target = 20，返回 false。

1、按照列表循环找出target

class Solution:
    def findNumberIn2DArray(self, matrix: List[List[int]], target: int) -> bool:
        for line in matrix:
            if target in line:
                return True
        return False

时间复杂度：O(nm)，空间复杂度O(1)

2、根据数组大小规律，进行查找

解题思路：

• 如果二维数组为空，直接return False
• i表示行，j表示列，每次都让target和右上角的数字matrix[i][j]比较
• 如果matrix[i][j]比target小，说明只需要在本行查找，也就是i行查找，让j左移，直到找出匹配的值
• 如果matrix[i][j]比target大，说明比这一行最大的数值还要大，那么就不需要在本行停留，让i下移，跳到下一行，然后与最右边数值进行比较
• 跳出循环的条件是i小于总行数，j大于0
• 如果最后都没有找到与target相等的值，直接return False

class Solution:
    def findNumberIn2DArray(self, matrix: List[List[int]], target: int) -> bool:
        m = len(matrix)
        if m == 0:
            return False
        n = len(matrix[0])
        i = 0
        j = n-1
        while i < m and j >= 0:
            cur = matrix[i][j]
            if cur > target:
                j -= 1
            if cur < target:
                i += 1
            if cur == target:
                return True
        return False

时间复杂度：O(n)，空间复杂度O(1)