【练习】【二分】力扣热题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;  // 返回结果
    }
};
相关推荐
不知天地为何吴女士2 小时前
Day32| 509. 斐波那契数、70. 爬楼梯、746. 使用最小花费爬楼梯
算法
小坏坏的大世界2 小时前
C++ STL常用容器总结(vector, deque, list, map, set)
c++·算法
liulilittle3 小时前
C++ TAP(基于任务的异步编程模式)
服务器·开发语言·网络·c++·分布式·任务·tap
励志要当大牛的小白菜4 小时前
ART配对软件使用
开发语言·c++·qt·算法
qq_513970444 小时前
力扣 hot100 Day56
算法·leetcode
PAK向日葵5 小时前
【算法导论】如何攻克一道Hard难度的LeetCode题?以「寻找两个正序数组的中位数」为例
c++·算法·面试
爱喝矿泉水的猛男7 小时前
非定长滑动窗口(持续更新)
算法·leetcode·职场和发展
YuTaoShao7 小时前
【LeetCode 热题 100】131. 分割回文串——回溯
java·算法·leetcode·深度优先
YouQian7728 小时前
Traffic Lights set的使用
算法
go54631584659 小时前
基于深度学习的食管癌右喉返神经旁淋巴结预测系统研究
图像处理·人工智能·深度学习·神经网络·算法