LeetCode 74 搜索二维矩阵 Python

最新推荐文章于 2021-09-07 09:34:21 发布

原创最新推荐文章于 2021-09-07 09:34:21 发布 · 206 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 同时被 2 个专栏收录

202 篇文章

订阅专栏

暑假算法

127 篇文章

订阅专栏

本文介绍了一种高效算法，用于在按特定规则排序的二维矩阵中查找目标值。矩阵的每行从左到右升序排列，且每行首元素大于前一行末元素。通过将矩阵视为一维有序数组，采用二分查找实现快速定位。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。该矩阵具有如下特性：

每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
示例 1:

输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
输出: true

之前做的二分查找的题都是一维的,但这道题是一个二维的矩阵，但是它的上一行的最后一个元素的值小于下一行的第一个元素的值，那么可以将下标看成一个一维已经被排序的数组。
从 0--m*n-1

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        if matrix == [[]] or matrix == []:
            return False
        m = len(matrix)
        n = len(matrix[0])
        l = 0
        r = m*n - 1
        while(l<r):
            mid = l + r >> 1
            y = mid // n 
            x = mid - n*y
            if matrix[y][x] >= target:
                r = mid
            else:
                l = mid + 1
        y = l // n
        x = l - n*y
        return matrix[y][x] == target