leetcode 34. 在排序数组中查找元素的第一个和最后一个位置（二分）

Posted 2021-03-03 wz-archer

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

你的算法时间复杂度必须是 O(log n) 级别。

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

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]
来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array

class Solution {
public:
    int lower_bound_(vector<int>& a,int x,int y,int v){
        int m;
        while(x<y){
            m=x+(y-x)/2;
            if(a[m]>=v)y=m;
            else x=m+1;
        }
        return x;
    }
    int lower_bound_2(vector<int>& a,int x,int y,int v){
        int m;
        while(x<y){
            m=x+(y-x)/2;
            if(a[m]>v)y=m;
            else x=m+1;
        }
        return x;
    }
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int>a;
        a.push_back(-1);
        a.push_back(-1);
        if(nums.size()==0)return a;
        int x=lower_bound_(nums,0,nums.size(),target);
        int y=lower_bound_2(nums,0,nums.size(),target);
        y--;
        if(x<nums.size()&&nums[x]==target&&x<=y){
            a[0]=x;
            a[1]=y;
        }
        return a;
    }
};