——Search for a Range

34、Search for a Range

搜索區間

給定一個包含 n 個整數的排序數組，找出給定目標值 target 的起始和結束位置。

如果目標值不在數組中，則返回[-1, -1]

樣例

給出[5, 7, 7, 8, 8, 10]和目標值target=8,

返回[3, 4]

分析：

先用二分查找找出第一個target，然後往後遍歷查找最後一個target。

代碼：

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> res(2,-1);
        if(nums.size()<1)return res;
        int lo=0,hi=nums.size()-1;
        while(lo<hi)
        {
            int mid=lo+(hi-lo)/2;
            if(nums[mid]<target)lo=mid+1;
            else hi=mid;
        }
        if(nums[lo]!=target)return res;
        while(hi+1<nums.size()&&nums[hi+1]==target)hi++;
        res[0]=lo;
        res[1]=hi;
        return res;
    }
};