正常的二分查找,多了重复元素的限制,另外要求重复元素的头尾位置。
于是在不经意间,我发现了这个秘密。
3、正文
一起来看一下是什么秘密?
首先序列是 5、7、7、8、10,target=7
,按照模板一二分查找。
int left=0;
int right=nums.size()-1;
while(left<right){
int mid=left right>>1;
if(nums[mid]>=target){
right=mid;
}
else{
left=mid 1;
}
}
return left;
可以看到,返回的是左侧的这个值,mid
是向下去整,right
也是,所以~
而模板二呢?
代码语言:javascript复制int left=0;
int right=nums.size()-1;
while(left<right){
int mid=(left right 1)>>1;
if(nums[mid]<=target){
left=mid;
}
else{
right=mid-1;
}
}
return right;
可以看到,返回的是右侧的这个值,mid
是向上去整,left
也是,所以~
最后返回的值写 left
或者 right
都是没问题的!因为 while
的结束条件是 left
和 right
相等。
4、代码
代码语言:javascript复制class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
if(nums.empty()) return {-1,-1};
int left=0;
int right=nums.size()-1;
while(left<right){
int mid=(left right)>>1;
if(nums[mid]>=target){
right=mid;
}
else{
left=mid 1;
}
}
if(nums[left]!=target) return {-1,-1};
int start=left;
left=0;
right=nums.size()-1;
while(left<right){
int mid=(left right 1)>>1;
if(nums[mid]<=target){
left=mid;
}
else{
right=mid-1;
}
}
if(nums[left]!=target) return {-1,-1};
int end=right;
return {start,end};
}
};