给你一个按照非递减顺序排列的整数数组 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]
思路来源:我写了首诗,让你闭着眼睛也能写对二分搜索 :: labuladong的算法小抄 (gitee.io)
左侧边界的二分查找、右侧边界的二分查找
class Solution {
public:
int leftfind(vector<int>&,int);
int rightfind(vector<int>&,int);
vector<int> searchRange(vector<int>& nums, int target)
{
int l=leftfind(nums,target);
int r=rightfind(nums,target);
vector<int> v;
v.push_back(l);
v.push_back(r);
return v;
}
};
int Solution::leftfind(vector<int> &nums,int target)
{
int left=0;
int right=nums.size();//采用[left,right)左闭右开的区间
int mid=0;
while(left<right)//当left==right时,[left,right)内正好没有元素
{
mid=left+(right-left)/2;
if(nums[mid]==target)//找到target时不要立即返回,而是缩小right,在区间 [left, mid) 中继续搜索,达到锁定左侧边界的目的
{
right=mid;
}
else if(nums[mid]>target)//在区间[left,mid)中搜索
{
right=mid;
}
else if(nums[mid]<target)//在区间[mid+1,right)中搜索
{
left=mid+1;
}
}
if(left==nums.size())//如果target比nums中的元素都大,left一直向右移,直到left=right=nums.size(),nums[left]产生数组越界
{
return -1;
}
return nums[left]==target?left:-1;//判断nums[left]是否等于target
}
int Solution::rightfind(vector<int> &nums,int target)
{
int left=0;
int right=nums.size();//[left,right)
int mid=0;
while(left<right)
{
mid=left+(right-left)/2;
if(nums[mid]==target)//找到target时不要立即返回,而是扩大left,在区间 [mid+1, right) 中继续搜索,达到锁定右侧边界的目的
{
left=mid+1;
}
else if(nums[mid]<target)//[mid+1,right)
{
left=mid+1;
}
else if(nums[mid]>target)//[left,mid)
{
right=mid;
}
}
if(left==0) return -1;//如果left==0,nums[left-1],会数组越界
return nums[left-1]==target?(left-1):-1;
}