题目描述:来源(LeetCode热题100第34题)
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
进阶:
你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗?
示例 1:
输入:nums = [5,7,7,8,8,10], target = 8
输出:[3,4]
示例 2:输入:nums = [5,7,7,8,8,10], target = 6
输出:[-1,-1]
示例 3:输入:nums = [], target = 0
输出:[-1,-1]
思路:
看到了logn不得不想到用二分,而且这刚好是一个排序数组,得来全不费工夫。
以前的二分可能只是用来查找某一个值在有序序列的位置,但这次是某个值在序列第一次和最后一次出现的位置,那我们可以先找出数组第一次出现的位置,再找出第二次出现的位置,两次用二分的不同就是判断区间的条件不同,找第一次出现的位置就是找到第一个大于等于目标值(判断继续二分的条件就是nums[mid]>=target 则让r=mid)的位置,找到最后一个出现的位置就是找到第一个小于等于目标值(判断继续二分的条件就是nums[mid]<=target 则让l=mid)的位置
代码实现C++:
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
int n=nums.size();
vector<int> res;
res.push_back(-1);//给vector数组初值
res.push_back(-1);
if(!n) return res;
int l=0,r=n-1;
while(l<r){//寻找第一次出现的位置,即第一个大于等于target的位置
int mid=(l+r)>>1;
if(nums[mid]>=target) r=mid;
else l=mid+1;
}
if(nums[l]!=target) return res;//退出while时l=r,如果nums[l]!=target,说明数组中不存在target,直接返回初始化的vector数组
else{//如果nums[l]=target,说明target找到了第一次出现的位置,就是l
res.pop_back();
res.pop_back();
res.push_back(l);
res.push_back(l);
l=0,r=n-1;
while(l<r){//查找最后一次出现的位置,即第一个小于等于target的位置
int mid=(l+r+1)/2;
if(nums[mid]<=target){
l=mid;
}else{
r=mid-1;
}
}
if(nums[l]==target){//如果退出循环的时候,nums[l]==target说明l就是最后一次出现的位置,否则就出现了一次,不会进入if
res.pop_back();
res.push_back(l);
}
}
return res;
}
};
二分模板:
bool check(int x){...};//检查x是否满足某种性质
// 区间[l, r]被划分成[l, mid]和[mid + 1, r]时使用(取mid左边)
int bsearch(int l,int r){
if(l==r) return a[l];
while(l<r){
int mid=l+r>>1;
if(check(mid)) r=mid;
else l=mid+1;
}
}
// 区间[l, r]被划分成[l, mid - 1]和[mid, r]时使用(取mid右边)
int bsearch(int l,int r){
if(l==r) return a[l];
while(l<r){
int mid=l+r+1>>1;
if(check(mid)) l=mid;
else r=mid-1;
}
}
今日刷题任务完成,明天见~(有疑问欢迎交流哦~)创作不易,谢谢支持!!