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,也不能停止,还要继续向左找更早的。

相关推荐
草莓熊Lotso2 小时前
《算法闯关指南:递归,搜索与回溯算法--递归》--02. 合并两个有序链表,03. 反转链表
运维·数据结构·算法·链表
XFF不秃头2 小时前
力扣刷题笔记-和为 K 的子数组
c++·笔记·算法·leetcode
MicroTech20252 小时前
区块链赋能,联邦协同:微算法科技(NASDAQ: MLGO)打造物联网安全分布式检测新架构
科技·算法·区块链
1001101_QIA2 小时前
vector学习笔记
数据结构·算法
(❁´◡`❁)Jimmy(❁´◡`❁)2 小时前
【数据结构】 Treap1: 插入,删除,查找,旋转
数据结构·算法
core5122 小时前
ResNet 残差连接:通往深层网络的“高速公路”
人工智能·算法·resnet
聆风吟º2 小时前
【数据结构手札】顺序表实战指南(五):查找 | 任意位置增删
数据结构·顺序表·查找·任意位置增删
曾几何时`2 小时前
滑动定窗口(十四)2831. 找出最长等值子数组
数据结构·算法
报错小能手2 小时前
数据结构 哈希基础 哈希函数 哈希冲突及解决
数据结构·哈希算法·散列表