给定一个按照升序排列的整数数组 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]
利用二分法,命中的时候分别找输出数组的左右边界
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> r(2,-1);
if(!nums.size()) return r;//为空时,返回[-1,-1]
int l = 0,h = nums.size()-1,m=0;
if(target<nums[l] || nums[h]<target) return r; //小于最小或大于最大返回[-1,-1]
while(l<=h){
m = (l+h)/2;
if(nums[m]==target){//命中m时
int i=m;
while(nums[i]==target&&i<nums.size()) ++i;//寻找nums中与target相等值的右界
r[1]=i-1;
while(nums[m]==target&&m>=0) --m;//寻找nums中与target相等值的左界
r[0]=m+1;
return r;
}
else if(nums[m]<target) l=m+1;
else h=m-1;
}
return r;//没找到target,返回[-1,-1]
}
};