给定一个按照升序排列的整数数组 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]
思路
- 先用二分搜索,找到index
- 再沿着index向两边搜索
code
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
int index = searchRange(nums,0,nums.size()-1,target);
int end = nums.size()-1;
vector<int> v;
if(index==-1)
return {-1,-1};
for(int i=index;i<=end;++i)
if(nums[i]!=target)
{
v.push_back(i-1);
break;
}
else if(i==end)
v.push_back(i);
for(int i=index;i>=0;--i)
if(nums[i]!=target)
{
v.push_back(i+1);
break;
}
else if(i==0)
v.push_back(i);
sort(v.begin(),v.end());
return v;
}
int searchRange(vector<int> nums,int start,int end,int target){
int mid = (start+end)/2;
if(start>end)
return -1;
if(nums[mid]==target)
return mid;
if(nums[mid]>target)
return searchRange(nums,start,mid-1,target);
else
return searchRange(nums,mid+1,end,target);
}
};