74. 搜索二维矩阵、240. 搜索二维矩阵 II

松语编程

于 2020-02-15 21:29:53 发布

阅读量151

点赞数

分类专栏： Leetcode

本文链接：https://blog.csdn.net/qq_27921205/article/details/104334689

版权

Leetcode 专栏收录该内容

107 篇文章 1 订阅

订阅专栏

74. 搜索二维矩阵

思路：把二维数组看做是一维数组，而且是有序的，使用二分查找。
时间复杂度O(log(mn))
在这里插入图片描述

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        def binarysearch(nums,left,right,target):
            while left<=right:
                mid = (left+right)//2
                midnum = nums[mid//n][mid%n]  # 注意
                if midnum == target:
                    return True
                elif midnum > target:
                    right = mid-1
                else:
                    left = mid+1
            return False

        if not matrix: return False
        m,n = len(matrix),len(matrix[0])
        return binarysearch(matrix,0,m*n-1,target)

在这里插入图片描述

240. 搜索二维矩阵 II

思路：左下角是当列最大的元素，当行最小的元素。从左下角出发去寻找，最远找到右上角。
时间复杂度O(m+n)

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