【问题】给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]。
示例 1: 输入: nums = [5,7,7,8,8,10], target = 8 输出: [3,4] 示例 2: 输入: nums = [5,7,7,8,8,10], target = 6 输出: [-1,-1]
【思路】在一个排序数组中查找某个元素的算法,我们很容易可以写出来寻找一个元素左边界的二分算法,即将target与mid相等时,将右边界r变为mid-1,向右移动即可!
那么怎么去查找某个元素的第一个和最后一个位置呢?很简单的思路,我们将左边界算法的查找目标target变化一下就可以了,第一次查找target,可以得到左边界的位置,第二次查找target+1,可能找到也可能找不到,但返回的都是最后位置+1的位置。
class Solution { public: vector<int> searchRange(vector<int>& nums, int target) { int idx1 = lower_bound(nums, target); int idx2 = lower_bound(nums, target+1)-1; if (idx1 < nums.size() && nums[idx1] == target) return {idx1, idx2}; else return {-1, -1}; } int lower_bound(vector<int>& nums, int target) { int l = 0, r = nums.size()-1; while (l <= r) { int mid = l+(r-l)/2; if (nums[mid] < target) l = mid+1; else r = mid-1; } return l; } };