hot100-54在排序数组中查找元素的第一个和最后一个位置

一、题目

给定一个非递减顺序排列的数组nums,和target,找出给定目标值在数组中的开始位置和结束位置,如果数组中不存在目标值target,返回[-1,-1]。时间复杂度O(log n)。

二、思路:

在有序数组中找重复元素的范围,就用两次二分:一次找左边界(≥ target),一次找右边界(≤ target)

利用两次二分查找 :第一次找第一个大于等于 target 的位置 (左边界),第二次找最后一个小于等于 target 的位置 (右边界);通过将 == 合并到 >=<= 条件中,确保即使找到目标值也继续向边界方向搜索,从而精准定位重复元素的范围,整体时间复杂度为 O(log n)。

三、代码

复制代码
class Solution {
    public int[] searchRange(int[] nums, int target) {
        int left = findLeft(nums, target);
        int right = findRight(nums, target);

        // 如果左边界 > 右边界,说明 target 不存在
        if (left > right) {
            return new int[]{-1, -1};
        }
        return new int[]{left, right};
    }

    // 找第一个 >= target 的位置(左边界)
    private int findLeft(int[] nums, int target) {
        int left = 0, right = nums.length - 1;
        int ans = nums.length; // 默认设为越界值(表示未找到)

        while (left <= right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] >= target) {
                ans = mid;       // 记录可能的左边界
                right = mid - 1; // 继续向左找更小的
            } else {
                left = mid + 1;  // 太小了,去右边
            }
        }
        return ans;
    }

    // 找最后一个 <= target 的位置(右边界)
    private int findRight(int[] nums, int target) {
        int left = 0, right = nums.length - 1;
        int ans = -1; // 默认设为 -1(表示未找到)

        while (left <= right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] <= target) {
                ans = mid;       // 记录可能的右边界
                left = mid + 1;  // 继续向右找更大的
            } else {
                right = mid - 1; // 太大了,去左边
            }
        }
        return ans;
    }
}

注:

if (nums[mid] >= target) 等于没有单独拿出来,是因为即使找到了 target,也不能停止,还要继续向左找更早的。

相关推荐
地平线开发者8 小时前
SparseDrive 模型导出与性能优化实战
算法·自动驾驶
董董灿是个攻城狮9 小时前
大模型连载2:初步认识 tokenizer 的过程
算法
地平线开发者9 小时前
地平线 VP 接口工程实践(一):hbVPRoiResize 接口功能、使用约束与典型问题总结
算法·自动驾驶
罗西的思考9 小时前
AI Agent框架探秘:拆解 OpenHands(10)--- Runtime
人工智能·算法·机器学习
HXhlx13 小时前
CART决策树基本原理
算法·机器学习
Wect13 小时前
LeetCode 210. 课程表 II 题解:Kahn算法+DFS 双解法精讲
前端·算法·typescript
颜酱14 小时前
单调队列:滑动窗口极值问题的最优解(通用模板版)
javascript·后端·算法
Gorway20 小时前
解析残差网络 (ResNet)
算法
拖拉斯旋风21 小时前
LeetCode 经典算法题解析:优先队列与广度优先搜索的巧妙应用
算法
Wect21 小时前
LeetCode 207. 课程表:两种解法(BFS+DFS)详细解析
前端·算法·typescript