Leetcode 34 在排序数组中查找元素的第一个和最后一个位置
问题重述
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-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]
示例 3:
输入:nums = [], target = 0 输出:[-1,-1]
思路
和c++里面的lower_bound和upper_bound,定义两个函数,分别求出lower坐标和upper坐标即可。具体需要如何计算左右端点坐标请看具体代码,我们选择的时候左开右闭的区间,区间不同 l 和 r 赋值会有略微不同
最终代码
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
if (nums.empty()) return vector<int>{-1, -1};
int lower = lower_bound(nums, target);
int upper = upper_bound(nums, target) - 1;
if (lower == nums.size() || nums[lower] != target) {
return vector<int>{-1, -1};
}
return vector<int>{lower, upper};
}
int lower_bound (vector<int> &nums, int target) {
int l = 0, r = nums.size(), mid;
while (l < r) {
mid = l + (r - l) / 2;
if (nums[mid] >= target) {//因为求得是l坐标所以必须是>=
r = mid;
} else {
l = mid + 1;
}
}
return l;
}
int upper_bound (vector<int> &nums, int target) {
int l = 0, r = nums.size(), mid;
while (l < r) {
mid = l + (r - l) / 2;
if (nums[mid] > target) {//求得是r坐标,所以不可以是>=
r = mid;
} else {
l = mid + 1;
}
}
return l;
}
};