https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/
题目
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值,返回 [-1, -1]。
- 你的算法时间复杂度必须是 O(log n) 级别。
示例
输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]
思路
二分
Left_bound
- nums[mid] < target,这时候 l = mid + 1,缩小左边范围
- nums[mid] > target,这时候 r = mid - 1,缩小右边范围
- nums[mid] == target,这时候 r = mid,因为我们寻找左边界的时候,当 mid 处值为 target,那么左边界必定小于等于 r ,由于可能就是r,所以不能减一,而是等待接下来缩小左边范围
- l == r 时退出循环,此时若 nums[l] == target,则左边界就是 l ,否则值不存在
Right_bound
- nums[mid] < target,这时候 l = mid + 1,缩小左边范围
- nums[mid] > target,这时候 r = mid - 1,缩小右边范围
- nums[mid] == target,这时候 l = mid + 1,我们找去找右边界右边的点,如果不动的话,可能陷入死循环,这时候找右边一个点,返回值时判断一下即可
- l == r 时退出循环,此时若 r 未越界并且值为 target,那么返回 r ,在此之上,如果 r - 1不越界,且 r - 1处值为 target,返回 r - 1,
代码
class Solution {
public:
int left_bound(vector <int>& nums, int target) {
int l = 0;
int r = nums.size() - 1;
while (l < r) {
int mid = (l + r) >> 1;
if (nums[mid] < target) {
l = mid + 1;
} else if (nums[mid] > target) {
r = mid - 1;
} else if (nums[mid] == target) {
r = mid;
}
// cout << " l:" << l;
// cout << " r:" << r << endl;
}
if (nums[r] == target) return r;
else return -1;
}
int right_bound(vector <int>& nums, int target) {
int l = 0;
int r = nums.size() - 1;
while (l < r) {
int mid = (l + r) >> 1;
if (nums[mid] < target) {
l = mid + 1;
} else if (nums[mid] > target) {
r = mid - 1;
} else if (nums[mid] == target) {
l = mid + 1;
}
// cout << " l:" << l;
// cout << " r:" << r << endl;
}
if (r < nums.size() && nums[r] == target) return r;
if (r && nums[r - 1] == target) return r - 1;
return -1;
}
vector<int> searchRange(vector<int>& nums, int target) {
vector <int> ans;
int n = nums.size();
if (!n || nums[0] > target || nums[n - 1] < target) return vector <int> ({-1,-1});
return vector <int> ({left_bound(nums, target), right_bound(nums, target)});
}
};