描述
给定一个按照升序排列的整数数组 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:
int searchLowBound(vector<int>& nums, int target) {
int low=0;;
int high=nums.size()-1;
while(low<=high) {
int mid=low+(high-low)/2;
if(nums[mid]<target) {
low=mid+1;
}
else{
high=mid-1;
}
}
if(low>=nums.size()||nums[low]!=target) {
return -1;
}
return low;
}
int searchHighBound(vector<int>& nums, int target) {
int low=0;;
int high=nums.size()-1;
while(low<=high) {
int mid=low+(high-low)/2;
if(nums[mid]>target) {
high=mid-1;
}
else {
low=mid+1;
}
}
if(high<0||nums[high]!=target) {
return -1;
}
return high;
}
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> res(2,-1);
res[0]=searchLowBound(nums,target);
res[1]=searchHighBound(nums,target);
return res;
}
};