力扣HOT100之二分查找: 34. 在排序数组中查找元素的第一个和最后一个位置
这道题完全没有思路,直接去看灵神的题解了,这道题的大概思路就是额外定义一个二分查找函数,对于给定数组nums
和目标值target
,返回第一个大于等于target
的元素下标,当返回的下标处的元素恰好等于target
时,该下标就是区间的开始位置,若不等于target
,则整个数组中都没有target
,返回[-1, -1]
即可。如果没有返回[-1, -1]
,则说明数组中至少有一个target
,我们需要进一步寻找结束位置。至于结束位置,我们直接调用该二分查找函数,查找target + 1
,返回的下标一定是第一个大于等于target + 1
的元素下标,不管数组中是否真的有target + 1
,此时下标的左边就是最后一个target
,因此我们直接返回起始坐标和结束坐标。
class Solution {
public:int Binary_search(vector<int>& nums, int target){int left = 0, right = nums.size(); //搜索区间为[left, right)int mid;while(left < right){mid = (left + right) / 2;if(nums[mid] >= target)right = mid;else left = mid + 1;}return left;}vector<int> searchRange(vector<int>& nums, int target) {int start = Binary_search(nums, target);if(start == nums.size() || nums[start] != target)return {-1, -1};int end = Binary_search(nums, target + 1) - 1;return {start, end};}
};