[LeetCode Python3] 240. Search a 2D Matrix II +二分查找

最新推荐文章于 2021-08-24 18:30:10 发布

原创最新推荐文章于 2021-08-24 18:30:10 发布 · 155 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode每日一题专栏收录该内容

172 篇文章

订阅专栏

这篇博客介绍了一个Python实现的二维矩阵搜索算法，利用了二分查找的方法。该算法在给定的矩阵中寻找目标值，通过递归地缩小搜索范围，提高了查找效率。在遇到目标值或超出矩阵边界时返回相应的结果。

240. Search a 2D Matrix II

二维版二分查找

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        m, n = len(matrix), len(matrix[0])

        def helper(rlo, rhi, clo, chi):
            if rlo >= rhi or clo >= chi:
                return False
            rmid, cmid = rlo + (rhi-rlo)//2, clo + (chi-clo)//2
            if matrix[rmid][cmid] == target:
                return True
            elif matrix[rmid][cmid] > target: # 矩阵的右下角也必大于target故排除
                return helper(rlo, rmid, clo, chi) or helper(rlo, rhi, clo, cmid)
            else:   # 矩阵的左上角也必小于target故排除
                return helper(rmid+1, rhi, clo, chi) or helper(rlo, rhi, cmid+1, chi)
        return helper(0, m, 0, n)