题目描述
Given an array of integers sorted in ascending order, find the starting and ending position of a given target value.
Your algorithm’s runtime complexity must be in the order of O(log n).
If the target is not found in the array, return [-1, -1]
.
For example,
Given [5, 7, 7, 8, 8, 10]
and target value 8,
return [3, 4]
.
解题思路
Binary Search。
题目要求O(log n)的时间复杂度,所以考虑使用二分查找。由于需要找整个范围的前后两个界限,所以考虑使用两次二分查找分别来找左右两个数字。
对于左边的界限,每次检索,考虑如下三种情况:
1. target > nums[mid], 那么往右继续查找
2. target < nums[mid], 那么往左继续查找
3. targe == nums[mid], 那么它有可能是左边界限,也有可能左界还在它的前面,所以继续往左找
这样,后两种情况可以合并。最后判断leftIdx停留在的位置是否为target,如果不是,说明target不存在数组中,也就不会有什么右界,直接return。否则,继续查找右界。
现在左界已经确定,只需从[leftIdx, nums.size()) 这个范围中查找右界。如同上面的分析,我们可以知道在找右界时,每次检索,有如下情况
1. target >= nums[mid], 往右查找
2. target < nums[mid], 往左查找
注意,由于int除法总是往下round,所以我们的midIdx永远是偏向于左边的,而当我们发现情况1时,会选择更新midIdx = leftIdx。那么,问题来了,如果出现了如下情况
[5, 8], target=5
算法就会陷入死循环,所以我们应该设法让midIdx右偏,让midIdx=(leftIdx+rightIdx)/2+1,这是非常关键的。
在二分查找中,考虑如下两种情况:
- 如果要使找到的值在最前面,那么,对于target==nums[mid],应该往左找。所以对于往右的情况,不需要在考虑中间值,只需设置leftIdx=nums[mid]+1,这样也不会受round左偏的影响。
- 如果要使找打的值在最后面,那么,对于target==nums[mid],应该往右找。由于在往右时,我们还是要考虑中间值的情况,所以设置leftIdx=nums[mid]。这就会出现上述的mid左偏死循环的情况,所以要给mid+1。
AC代码
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> ans;
ans.push_back(-1);
ans.push_back(-1);
if (nums.empty())
return ans;
const int vSize = nums.size();
//find the start index
int leftIdx = 0, rightIdx = vSize - 1;
int midIdx;
while (leftIdx < rightIdx) {
midIdx = (leftIdx + rightIdx) / 2;
if (target > nums[midIdx]) {
leftIdx = midIdx + 1;
}
else {
rightIdx = midIdx;
}
}
if (nums[leftIdx] == target)
ans[0] = leftIdx;
else
return ans;
//find the end index, start from the leftIdx found above
rightIdx = vSize - 1;
while (leftIdx < rightIdx) {
midIdx = (leftIdx + rightIdx) / 2 + 1; // Make mid bais to right
if (target < nums[midIdx])
rightIdx = midIdx - 1;
else
leftIdx = midIdx;
}
ans[1] = leftIdx;
return ans;
}
};