二分查找:左右边界

文章目录

一、把问题拆成两次边界搜索

LeetCode 34. 在排序数组中查找元素的第一个和最后一个位置

左边界是第一个等于 target 的位置,右边界是最后一个等于 target 的位置。使用闭区间 [left, right] 搜索时,需要先确定区间外已经知道什么,再决定 mid 是否仍可能是答案。

搜索目标 left 左侧 right 右侧
左边界 < target >= target
右边界 <= target > target

表格描述的是候选区间外侧,不包含仍待判断的 [left, right]

二、寻找左边界

左边界搜索使用下中点:

text 复制代码
mid = left + (right - left) / 2
  • nums[mid] < target 时,mid 及其左侧不可能是答案,令 left = mid + 1
  • nums[mid] >= target 时,mid 仍可能是第一个目标值,令 right = mid 保留它。

当区间只剩 [i, i + 1] 时,下中点为 i。无论更新 left = mid + 1 还是 right = mid,区间都会继续缩小。

三、寻找右边界

右边界搜索使用上中点:

text 复制代码
mid = left + (right - left) / 2 + 1
  • nums[mid] <= target 时,mid 仍可能是最后一个目标值,令 left = mid 保留它。
  • nums[mid] > target 时,mid 及其右侧不可能是答案,令 right = mid - 1

如果这里仍使用下中点,那么在 [i, i + 1] 中会得到 mid == left == i。一旦进入 left = mid 分支,区间不会变化,从而产生死循环。改用上中点后,mid == i + 1,两个分支都能让区间收缩。

四、目标不存在时的校验

这两次搜索都会返回数组中的一个候选下标,但候选值不一定等于 target。例如数组只有 [5] 时,查找 46 都会得到候选位置 0,因此不能只用"左候选是否大于右候选"判断目标是否存在。

完整代码需要额外处理空数组,并检查两个候选位置的值:

java 复制代码
class Solution {
    public int[] searchRange(int[] nums, int target) {
        int[] ret = new int[2];
        int left = 0;
        int right = nums.length - 1;

        while (left < right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] < target) {
                left = mid + 1;
            } else {
                right = mid;
            }
        }
        ret[0] = left;

        left = 0;
        right = nums.length - 1;
        while (left < right) {
            int mid = left + (right - left) / 2 + 1;
            if (nums[mid] <= target) {
                left = mid;
            } else {
                right = mid - 1;
            }
        }
        ret[1] = right;

        if (nums.length == 0
                || nums[ret[0]] != target
                || nums[ret[1]] != target) {
            ret[0] = -1;
            ret[1] = -1;
        }
        return ret;
    }
}

两次二分搜索的时间复杂度均为 O(log n),总时间复杂度仍为 O(log n);除返回数组外,额外空间复杂度为 O(1)

相关推荐
lazily-c2 小时前
数据结构模板
数据结构
祖力5510 小时前
数据结构的基本概念与单向链表(链表数据类型构造、创建、头插、尾插、头删、尾删)
数据结构·链表
大明者省11 小时前
WSL2 Ubuntu22.04 GPU训练环境配置指南
人工智能·算法·计算机视觉
白狐_79813 小时前
408数据结构第8章:排序②——性质对比秒杀、场景选择与外部排序
java·数据结构·算法
zander25813 小时前
LeetCode 84:柱状图中的最大矩形——单调栈如何确定左右边界
java·数据结构·算法
Shell运维手记13 小时前
Linux 常用基础命令学习笔记
linux·运维·笔记·学习·算法·github
杨航 AI14 小时前
O(n log n):线性对数原理拆解 这个是排序算法的黄金复杂度之一。
算法·排序算法
船厂电气自动化ai大模型14 小时前
AI大模型与数学 第32课 函数凹凸性与二阶导数:拐点求解、凹凸区间计算(10道二阶导数计算题)
数据结构·人工智能·python·深度学习·算法
旖旎夜光14 小时前
LeetCode 3:无重复字符的最长子串(滑动窗口) —— 题解
数据结构·c++·算法·leetcode·滑动窗口