给定一个按照升序排列的整数数组 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]
思路:
这道题两个关键点:数字是有序的,时间复杂度为O(logn)。
题目可以转化为两个子问题,1求排序数组中出现的第一个等于target的数字下标,2求排序数组中出现的最后一个等于target的数字下标。
具体两个字问题参考这两篇博文(第一个大于等于target的数字,最后一个等于target的数字之类都属于二分查找的变形版,应该牢记常见的变形版本和解答方法)
https://blog.csdn.net/ebowtang/article/details/50770315
循环不变式下的二分法,http://www.cnblogs.com/wuyuegb2312/archive/2013/05/26/3090369.html
求出下标以后,直接填写到res中即可,如果没有找到就填写-1
代码如下:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> res;
if(nums.size()<=0){
return {-1,-1};
}
int left = 0;
int right = nums.size() - 1;
while (left < right) {
int mid = left + ((right - left) >> 1);
if (nums[mid] < target) {
left = mid + 1;
}
else {
right = mid;
}
}
if (nums[left] == target) {
res.push_back(left);
}
else {
res.push_back(-1);
}
left = 0;
right= nums.size() - 1;
while (left < (right-1)) {
int mid = left + ((right - left) >> 1);
if (nums[mid] > target) {
right = mid - 1;
}
else {
left = mid;
}
}
if (nums[left] <= target && nums[right] == target) {
res.push_back(right);
}
else if (nums[left] == target && nums[right] > target) {
res.push_back(left);
}
else {
res.push_back(-1);
}
return res;
}