在排序数组中查找元素的第一个和最后一个位置
给定一个按照升序排列的整数数组 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]
二分查找
先排序,m=nums.size(),对于升序数组l=0,r=m-1
mid=(l+r)/2
如果mid>target 则r=mid-1 之后更新mid
如果mid<target 则l=mid+1 之后更新mid
重复上述过程 直到nums[mid]==target(找到)l>r(没找到)
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
int l=0;int m=nums.size();int r=m-1;vector<int> an;
int mid=(l+r)/2;int first,last;
if(m==1)
{
if(nums[0]==target)
{
return {0,0};
}
else{
return {-1,-1};
}
}
if(m==0)
{
return {-1,-1};
}
while(mid<m&&nums[mid]!=target&&l<r)
{
if(nums[mid]>target)
{
r=mid-1;
}
else if(nums[mid]<target)
{
l=mid+1;
}
mid=(l+r)/2;
}
if(nums[mid]!=target)
{
an.push_back(-1);
an.push_back(-1);
return an;
}
int i=mid;
while(i<m&&nums[i]==target)
{
i++;
}
last=i-1;i=mid;
while(i>=0&&nums[i]==target)
{
i--;
}
first=i+1;
an.push_back(first);an.push_back(last);
return an;
}
};
本题二分搜索比较简单
但是细节很重要
比如
[]
0;
[1]
1;
所以一定注意特殊情况细节