240. 搜索二维矩阵 II Python

哭惹哭惹2023-09-24 19:34

文章目录

一、题目描述
- - [示例 1](#示例 1)
  - [示例 2](#示例 2)
二、代码
三、解题思路

一、题目描述

编写一个高效的算法来搜索 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
-10^9 <= matrix[i][j] <= 10^9
每行的所有元素从左到右升序排列
每列的所有元素从上到下升序排列
-10^9 <= target <= 10^9

二、代码

代码如下：

c 复制代码

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

三、解题思路

本题是对于特殊矩阵的遍历操作，由于该矩阵的每一行和每一列都满足升序，则可以先遍历行来找出目标值可能的存在的行数，然后再到行里面具体寻找每一列是否存在目标值，如果出现当前值已经比目标值大的情况，则表示后面的序列都不可能出现目标值了，直接结束当前循环；如果找到目标值，则立刻停止遍历，返回结果即可。