随机一题给定一个按照升序排列的整数数组 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]
思路:这里使用2次二分查找来获取目标值的头尾位置,如果头的位置大于尾的位置则不存在这个元素。第一次二分查找的结果是lower是第一个目标值的索引,第二次查找的结果upper是最后一个目标值的索引,最后将其返回。
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
int nCount = nums.size();
int lower,upper;
int i=0,j=nCount-1;
while(i<=j)
{
int Mid = (i+j)/2;
if(nums[Mid] < target)
i = Mid+1;
else
j = Mid-1;
}
lower = i;
i=0,j=nCount-1;
while(i<=j)
{
int Mid = (i+j)/2;
if(nums[Mid] <= target)
i = Mid+1;
else
j = Mid-1;
}
upper = j;
if(lower > upper)
return {-1,-1};
else
return {lower,upper};
}
};