leetcode -- Search a 2D Matrix -- 重点--BS

最新推荐文章于 2024-11-04 12:20:00 发布

xyqzki

最新推荐文章于 2024-11-04 12:20:00 发布

阅读量441

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/xyqzki/article/details/50320889

版权

leetcode 专栏收录该内容

257 篇文章 0 订阅

订阅专栏

本文介绍了如何使用两次二分查找法解决LeetCode中的二维矩阵搜索问题，达到O(logm + logn)的时间复杂度。同时提供了两种不同的解题思路，包括直接在矩阵上进行搜索的实现方式。

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

https://oj.leetcode.com/problems/search-a-2d-matrix/

思路1

思路就是做两次BS. 第一次定位行，第二次定位列
O(logm + logn)
http://blog.csdn.net/linhuanmars/article/details/24216235

python code
http://jelices.blogspot.hk/2014/05/leetcode-python-search-2d-matrix.html

自己重写code

class Solution(object):
    def searchMatrix(self, matrix, target):
        """
        :type matrix: List[List[int]]
        :type target: int
        :rtype: bool
        """
        m, n = len(matrix), len(matrix[0])
        low = 0
        high = m - 1

        while low <= high:
            mid = (low + high) / 2
            if matrix[mid][0] == target:
                return True
            elif matrix[mid][0] < target:
                low = mid + 1
            else:
                high = mid - 1
        low -= 1
        x = low
        low, high = 0, len(matrix[x]) - 1

        while low <= high:
            mid = (low + high) / 2
            if matrix[x][mid] == target:
                return True
            elif matrix[x][mid] < target:
                low = mid + 1
            else:
                high = mid - 1
        return False

思路2 直接在matrix上搜索

http://www.cnblogs.com/zuoyuan/p/3770061.html

class Solution:
    # @param matrix, a list of lists of integers
    # @param target, an integer
    # @return a boolean
    def searchMatrix(self, matrix, target):
        i = 0; j = len(matrix[0]) - 1
        while i < len(matrix) and j >= 0:
            if matrix[i][j] == target: return True
            elif matrix[i][j] > target: j -= 1
            else: i += 1
        return False