【练习】【二分】力扣热题100 34. 在排序数组中查找元素的第一个和最后一个位置

题目

给你一个按照非递减顺序排列的整数数组 nums,和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target,返回 [-1, -1]。

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。
示例 1:

输入:nums = [5,7,7,8,8,10], target = 8

输出:[3,4]
示例 2:

输入:nums = [5,7,7,8,8,10], target = 6

输出:[-1,-1]
示例 3:

输入:nums = [], target = 0

输出:[-1,-1]

来源:力扣热题100 34. 在排序数组中查找元素的第一个和最后一个位置


思路(注意事项)


纯代码

c 复制代码
class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> ans;
        int n = nums.size();
        if (n == 0) return {-1, -1};
        int l = 0, r = n - 1;
        while (l < r)
        {
            int mid = l + r >> 1;
            if (nums[mid] >= target) r = mid;
            else l = mid + 1;
        }
        if (nums[r] == target) ans.push_back(r);
        else 
        {
            ans.push_back(-1);
            ans.push_back(-1);
            return ans;
        }

        l = 0;
        r = n - 1;
        while (l < r)
        {
            int mid = l + r + 1 >> 1;
            if (nums[mid] <= target) l = mid;
            else r = mid - 1;
        }
        ans.push_back(r);

        return ans;
    }
};

题解(加注释)

c 复制代码
class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> ans;  // 存储结果的数组
        int n = nums.size();  // 数组的长度

        // 如果数组为空,直接返回 {-1, -1}
        if (n == 0) return {-1, -1};

        // 第一次二分查找:查找 target 的起始位置
        int l = 0, r = n - 1;  // 初始化左右指针
        while (l < r) {
            int mid = l + r >> 1;  // 计算中间位置
            if (nums[mid] >= target) r = mid;  // 如果中间值大于等于 target,缩小右边界
            else l = mid + 1;  // 否则缩小左边界
        }

        // 检查是否找到 target
        if (nums[r] == target) ans.push_back(r);  // 如果找到,将起始位置加入结果
        else {
            ans.push_back(-1);  // 如果没找到,返回 {-1, -1}
            ans.push_back(-1);
            return ans;
        }

        // 第二次二分查找:查找 target 的结束位置
        l = 0;  // 重置左指针
        r = n - 1;  // 重置右指针
        while (l < r) {
            int mid = l + r + 1 >> 1;  // 计算中间位置(注意 +1 防止死循环)
            if (nums[mid] <= target) l = mid;  // 如果中间值小于等于 target,缩小左边界
            else r = mid - 1;  // 否则缩小右边界
        }

        // 将结束位置加入结果
        ans.push_back(r);

        return ans;  // 返回结果
    }
};
相关推荐
Touper.14 分钟前
JavaSE -- 泛型详细介绍
java·开发语言·算法
sun00770017 分钟前
std::forward作用
开发语言·c++·算法
JoernLee35 分钟前
机器学习算法:支持向量机SVM
人工智能·算法·机器学习
V我五十买鸡腿43 分钟前
顺序栈和链式栈
c语言·数据结构·笔记·算法
海棠玛卡1 小时前
C/C++内存管理
c++
我爱一条柴ya1 小时前
【AI大模型】线性回归:经典算法的深度解析与实战指南
人工智能·python·算法·ai·ai编程
虾球xz1 小时前
CppCon 2018 学习:THE MOST VALUABLE VALUES
开发语言·c++·学习
三维重建-光栅投影3 小时前
VS中将cuda项目编译为DLL并调用
算法
2401_881244403 小时前
牛客周赛99
c++
课堂剪切板5 小时前
ch03 部分题目思路
算法