【leetcode】剑指 Offer 04. 二维数组中的查找（python）

最新推荐文章于 2022-10-25 14:56:21 发布

zzamx

最新推荐文章于 2022-10-25 14:56:21 发布

阅读量227

点赞数

分类专栏：每日打卡 Python Leetcode 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/xiaoyue_/article/details/125298745

版权

Leetcode 同时被 3 个专栏收录

119 篇文章 1 订阅

订阅专栏

Python

70 篇文章 0 订阅

订阅专栏

每日打卡

18 篇文章 0 订阅

订阅专栏

在这里插入图片描述

方法一：设置左下角元素为标记flag，依次消去每行每列(左下角->右上角)

时间复杂度：O(m + n)
详细请参考题解

# 题解，时间复杂度：O(m + n)
class Solution:
    def findNumberIn2DArray(self, matrix, target) -> bool:
        i, j = len(matrix) - 1, 0
        while i >= 0 and j < len(matrix[0]):
            flag = matrix[i][j]
            if target > flag: # 消去当前列
                j += 1
            elif target < flag: # 消去当前行  
                i -= 1
            else:
                return True
        return False

方法二：遍历每行，每行数组用二分查找

时间复杂度：O(nlogn)

class Solution:
    def findNumberIn2DArray(self, matrix: List[List[int]], target: int) -> bool:
        if not matrix or not matrix[0]:
            return False
        m = len(matrix[0])
        for row in matrix:
            if target >= row[0] and target <= row[m - 1]:
                l = 0
                r = m - 1
                while l <= r:   ## 细节！ not < 
                    mid = (l + r ) // 2
                    if target == row[mid]:
                        return True
                    elif target < row[mid]:
                        r = mid - 1
                    else:
                        l = mid + 1
        return False