题目

240. 搜索二维矩阵 II

编写一个高效的算法来搜索 *m* x *n* 矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性：

• 每行的元素从左到右升序排列。
• 每列的元素从上到下升序排列。

示例 1：

输入：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

示例 2：

输入：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 = 20
输出：false

提示：

• m == matrix.length
• n == matrix[i].length
• 1 <= n, m <= 300
• -109 <= matrix[i][j] <= 109
• 每行的所有元素从左到右升序排列
• 每列的所有元素从上到下升序排列
• -109 <= target <= 109

思路

• 将初始点设置在左下角（右上同理），大则右移，小则左移

代码

	from typing import List


class Solution:
    def findNumberIn2DArray1(self, matrix: List[List[int]], target: int) -> bool:
        for i in range(len(matrix)):
            for j in range(len(matrix[0])):
                if matrix[i][j] == target:
                    return True
        return False

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

复杂度

• 时间复杂度：$O(n+m)$
• 空间复杂度：$O(1)$