LeetCode第 34 题:在排序数组中查找元素的第一个和最后一个位置(C++)

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

就是二分查找的变体,感觉跟简单题差不多:

数据结构与算法之美:16 | 二分查找(下)

这儿怎么找都行,那就找第一个等于target的位置吧:

class Solution {
public:
    int bsearch(vector<int> &nums, int low, int high, int target){
        while(low <= high){
            auto mid = (low+high)>>1;
            if(nums[mid] > target)  high = mid-1;
            else if(nums[mid] < target) low = mid+1;
            else{
                if(mid == 0 || nums[mid-1] != target)   return mid;
                high = mid-1;
            }
        }
        return -1;
    }
    vector<int> searchRange(vector<int>& nums, int target) {
        int index = bsearch(nums, 0, nums.size()-1, target);
        if(index == -1) return {-1,-1};
        int i = index+1;
        while(i < nums.size() && nums[i] == nums[index])    ++i;
        return {index, i-1};
    }
};

也可以两次二分查找,第二次找最后一个就行。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值