题目要求时间复杂度为O(logn),所以要采用二分查找,但是直接采用二分查找可能存在将待查找的数分到左右两个序列中,无法解决。所以采用两次二分查找查找开始位置和结束位置。但是需要确认的是,我们寻找的开始位置和结束位置不能将target跳过。所以当nums[left]小于(没有等于,确保不会将target漏过)target时,开始位置更新为left+1,同理,当nums[right]大于(没有等于,确保不会将target漏过)target时,结束位置更新为right-1。这样的查找为完整的target序列。
代码如下,具体解释见代码注释。
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
//nums为空,直接返回{-1,-1}
if (nums.empty()) {
return { -1,-1 };
}
vector<int> vec(2);
int l = 0, r = nums.size() - 1;
while (l < r) {
//查找开始位置的时候确保中点要偏左,所以mid=l+(r-l)/2
int mid = l + (r - l) / 2;
if (nums[mid] >= target) {
r = mid;
}
//当nums[mid] < target的时候才更新开始位置,这样可以确保target不会被跳过
else {
l = mid + 1;
}
}
//未找到直接返回{-1,-1}
if (nums[l] != target) {
return{ -1,-1 };
}
vec[0]=l;
l = 0, r = nums.size() - 1;
while (l < r) {
//查找结束位置的时候确保中点要偏右,所以mid=l+(r-l)/2+1
int mid = l + (r - l) / 2 + 1;
if (nums[mid] <= target) {
l = mid;
}
//当nums[mid] > target的时候才更新结束位置,这样可以确保target不会被跳过
else {
r = mid - 1;
}
}
if (nums[r] != target) {
return{ -1,-1 };
}
vec[1]=r;
return vec;
}
};