[每日一题][找工作第59天][2025-11-20] Leetcode 74. 搜索二维矩阵

题目

74. 搜索二维矩阵
给你一个满足下述两条属性的 m x n 整数矩阵:

  • 每行中的整数从左到右按非严格递增顺序排列。
  • 每行的第一个整数大于前一行的最后一个整数。
    给你一个整数 target ,如果 target 在矩阵中,返回 true ;否则,返回 false 。
     
    示例 1:

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

    输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13
    输出:false
     
    提示:
  • m == matrix.length
  • n == matrix[i].length
  • 1 <= m, n <= 100
  • -104 <= matrix[i][j], target <= 104

我的代码(Python)

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        m, n = len(matrix), len(matrix[0])
        if matrix[0][0] > target or matrix[m - 1][n - 1] < target:
            return False
        if matrix[m - 1][n - 1] == target or matrix[m - 1][0] == target:
            return True
        # row
        rs, re = 0, m - 1
        while rs < re:
            # print(f'rs: {rs}, re: {re}')
            # if matrix[rs][0] == target or matrix[re][0] == target:
            #     return True
            # if matrix[rs][0] > target:
            #     return False
            if matrix[re][0] < target:
                rs = re
                break
            if re - rs == 1:
                break
            rmid = (rs + re) // 2
            if matrix[rmid][0] == target:
                return True
            if matrix[rmid][0] < target:
                rs = rmid
            if matrix[rmid][0] > target:
                re = rmid
        r = rs

        # column
        cs, ce = 0, n - 1
        if matrix[r][cs] == target or matrix[r][ce] == target:
            return True
        while cs < ce:
            # if matrix[r][cs] == target or matrix[r][ce] == target:
            #     return True
            if matrix[r][cs] > target or matrix[r][ce] < target:
                return False
            if ce - cs == 1:
                break
            cmid = (cs + ce) // 2
            if matrix[r][cmid] == target:
                return True
            if matrix[r][cmid] < target:
                cs = cmid
            if matrix[r][cmid] > target:
                ce = cmid
        return matrix[r][cs] == target

点评

实际上就是一个递增序列,只是放成矩阵形式。
比起240. 搜索二维矩阵 II不是一个思路,简化很多。
我的解法
我的做法是先二分查找行,确定在哪一行再二分查找列。
也可以用整个数组的长度来做二分查找,然后每个索引计算它实际的行列数。
两者时间复杂度应该是一样的。
官方两种解法就是这个思路。


baddif@gmail.com

AI简历优化站

Nonpareil.me:优化简历,助力职场
开源代码

AI求职跟踪器(建设中)

开源代码

主站(建设中)

Nonpareil Me

相关平台

Github Issues / Notion / Blog

[每日一题][找工作第58天][2025-11-19] Leetcode 35. 搜索插入位置

题目

35. 搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
 
示例 1:
输入: nums = [1,3,5,6], target = 5
输出: 2
示例 2:
输入: nums = [1,3,5,6], target = 2
输出: 1
示例 3:
输入: nums = [1,3,5,6], target = 7
输出: 4
 
提示:

  • 1 <= nums.length <= 104
  • -104 <= nums[i] <= 104
  • nums 为 无重复元素 的 升序 排列数组
  • -104 <= target <= 104

我的代码(Python)

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        s, e = 0, len(nums) - 1
        if nums[s] >= target:
            return s
        if nums[e] == target:
            return e
        if nums[e] < target:
            return e + 1
        while s < e:
            if s == e - 1:
                return e
            m = (s + e) // 2
            if nums[m] == target:
                return m
            elif nums[m] < target:
                s = m
            else:
                e = m
        return s

点评

简单粗暴的二分查找,上下限索引的正确迭代和特殊情况的处理是关键点。
我比官方解法多做了一些特殊情况的代码处理,不过相对的效率会高一些。


baddif@gmail.com

AI简历优化站

Nonpareil.me:优化简历,助力职场
开源代码

AI求职跟踪器(建设中)

开源代码

主站(建设中)

Nonpareil Me

相关平台

Github Issues / Notion / Blog

发表评论

您的邮箱地址不会被公开。 必填项已用 * 标注

滚动至顶部