剑指offer leetcode python版 面试题04. 二维数组中的查找

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

显然，对矩阵进行简单的二维暴力遍历可以解决，但是没有利用矩阵中元素的大小关系，运行时间会超出限制。

由题目可知，矩阵元素大小大概是左上小于右下。那我们可以选择一个开始遍历的起点，当等于目标值的时候返回真，小于目标值的时候往下遍历，大于目标值的时候往左遍历。显然，这个遍历的起点在右上角。同理，遍历的起点也可以选择在左下角。这样一次遍历就可以跳过一行或者是一列的元素，节省了遍历时间。

python版。

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