在排序数组中查找元素的第一个和最后一个位置
给你一个按照非递减顺序排列的整数数组nums
,和一个目标值 target
。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target
,返回 [-1, -1]
。
你必须设计并实现时间复杂度为 O(log n)
的算法解决此问题。
示例 1:
输入:nums = [5,7,7,8,8,10]
, target = 8 输出:[3,4]
解题思路:
- 1 二分法寻找目标值
- 2 while循环左右寻找边界目标值
class Solution {
public:
int binarySearch(vector<int>& nums, int target){
int left = 0, right = nums.size()-1;
while(left <= right){
int middle = left + (right - left) / 2;
if (nums[middle] == target) return middle;
else if (nums[middle] < target) left = middle + 1;
else if (nums[middle] > target) right = middle - 1;
}
return -1;
}
vector<int> searchRange(vector<int>& nums, int target) {
int result = binarySearch(nums, target);
int left = result, right = result;
if (result == -1) return {-1, -1};
else{
while(left >= 0 && nums[left] == nums[result]){
left--;
}
while(right < nums.size() && nums[right] == nums[result]){
right++;
}
}
return {left+1, right-1};
}
};