给定一个按照升序排列的整数数组 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:
vector<int> searchRange(vector<int>& nums, int target)
{
vector<int> res(2,-1); // res[2] 用-1初始化
int left=0;
int right=nums.size()-1;
while(left<=right)
{
int mid = (left+right)/2;
if(nums[mid]<target)
{
left = mid+1; // 先动左端,所以查找到的是第一个
}
else
{
right=mid-1;
if(nums[mid]== target)
{
res[0] = mid;
}
}
}
left=0;
right=nums.size()-1;
while(left<=right)
{
int mid = (left+right)/2;
if(nums[mid]>target)
{
right = mid-1; // 先动右端,所以查找到的是最后一个
}
else
{
left = mid+1;
if(nums[mid]==target)
{
res[1] = mid;
}
}
}
return res;
}
};