给定一个按照升序排列的整数数组 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]
分析:
先使用二分查找,找到 target 的下标,然后往两边遍历,得到出现这个数字范围
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> result(2);
// 二分查找
int mid = binary_search(nums,0,nums.size()-1,target);
// 失败返回
if(mid == -1){
result[0] = result[1] = -1;
return result;
}
// 查找成功,继续往两边搜索
int left, right;
left = right = mid;
// 往两边搜索这个数字
while(left >= 0){
if(nums[left] != target)
break;
left --;
}
while(right < nums.size()){
if(nums[right] != target)
break;
right ++;
}
result[0] = left + 1;
result[1] = right - 1;
return result;
}
// 二分查找,返回下标
int binary_search(vector<int>& nums, int start, int end, int target){
int mid;
while(start <= end){
mid = (start + end)/2;
if(nums[mid] == target){
return mid;
}else if(nums[mid] < target){
start = mid + 1;
}else{
end = mid - 1;
}
}
return -1;
}
};