34. 在排序数组中查找元素的第一个和最后一个位置 - 力扣(LeetCode)
就是二分查找的变体,感觉跟简单题差不多:
这儿怎么找都行,那就找第一个等于target的位置吧:
class Solution {
public:
int bsearch(vector<int> &nums, int low, int high, int target){
while(low <= high){
auto mid = (low+high)>>1;
if(nums[mid] > target) high = mid-1;
else if(nums[mid] < target) low = mid+1;
else{
if(mid == 0 || nums[mid-1] != target) return mid;
high = mid-1;
}
}
return -1;
}
vector<int> searchRange(vector<int>& nums, int target) {
int index = bsearch(nums, 0, nums.size()-1, target);
if(index == -1) return {-1,-1};
int i = index+1;
while(i < nums.size() && nums[i] == nums[index]) ++i;
return {index, i-1};
}
};
也可以两次二分查找,第二次找最后一个就行。