这道题题目很简单,但是要保证O(log n)的时间复杂度,还是要花一些心思。本人实在水平有限,一开始就写了个超时的做法,连第一个样例都过不了。。。。看了答案的标题,用两次二分查找分别找头和尾,才有了思路。答案的代码很简洁,这里还是讲讲我自己的做法。先找头的位置,用二分查找,如果找到mid = target,那么头就在这个mid的左边,或者就是mid,如果mid左边的值小于mid处的值,则这就是要找的头部,否则,头部还在左边,我们让high = mid - 1然后继续查找就可以。查找尾部的方法相似。
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> results(2, -1);
int low = 0, high = nums.size() - 1;
while(low <= high){
int mid = (low + high) / 2;
if(nums[mid] == target){
if(mid == 0 || nums[mid - 1] < target){
results[0] = mid; break;
}
else high = mid - 1;
}
else if(target < nums[mid]) high = mid - 1;
else low = mid + 1;
}
low = 0; high = nums.size() - 1;
while(low <= high){
int mid = (low + high) / 2;
if(nums[mid] == target){
if(mid == nums.size() - 1 || nums[mid + 1] > target){
results[1] = mid; break;
}
else low = mid + 1;
}
else if(target < nums[mid]) high = mid - 1;
else low = mid + 1;
}
return results;
}