搜索二维矩阵

胡萝北北北北北北

于 2021-10-25 14:25:37 发布

阅读量87

点赞数

分类专栏：每天一个Python小技巧文章标签：矩阵 leetcode python

本文链接：https://blog.csdn.net/Ccarrot_w/article/details/120951139

版权

每天一个Python小技巧专栏收录该内容

33 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何解决LeetCode中的搜索二维矩阵问题。给定一个升序排列的矩阵，目标是高效地查找目标值。策略是从矩阵的右下角开始，根据目标值与当前值的比较，向上或向左移动，直到找到目标值或搜索完整个矩阵。

摘要由CSDN通过智能技术生成

LeetCode每日一题（2021.10.25）【MID】

编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性：
每行的元素从左到右升序排列。
每列的元素从上到下升序排列。
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-a-2d-matrix-ii

示例 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

由于是从左到右的升序，最右列的数一定大于左边的数，那么从最右列开始查找，如果大于target，则往左移一列；
同理，最下方的数一定大于上方的数，如果当前行的数小于target，那么行数加一。
直到查找到最左下角的数。

class Solution:
    def searchMatrix(self, matrix, target):
        m,n = len(matrix),len(matrix[0])
        row, col = 0, n-1
        while col >= 0 and row <= m-1:
            if matrix[row][col] > target:
                col -= 1
            elif matrix[row][col] < target:
                row += 1
            elif matrix[row][col] == target:
                return True
        return False