LeetCode34 在排序数组中查找元素

链接:https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/

题面

给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

这道题目可以说是直接告诉你要用二分来做,那么具体来说就是会用到C++里的lower_boundupper_bound两个函数。这里尝试自己实现这两个函数

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
class Solution {
public:
int lower_bd(vector<int> &nums, int target) {
int l = 0, r = nums.size();
while(l < r) {
int mid = l + (r - l) / 2;
if (nums[mid] >= target) {
r = mid;
}
else {
l = mid + 1;
}
}
return l;
}
int upper_bd(vector<int> &nums, int target) {
int l = 0, r = nums.size();
while(l < r) {
int mid = l + (r - l) / 2;
if (nums[mid] > target) {
r = mid;
}
else {
l = mid + 1;
}
}
return l;
}
vector<int> searchRange(vector<int>& nums, int target) {
if (nums.empty()) {
return {-1, -1};
}
int lower = lower_bd(nums, target);
int upper = upper_bd(nums, target) - 1;
if (upper < 0 || lower == nums.size() || nums[upper] != target) {
return {-1, -1};
}
else
return {lower, upper};
}

};

当然也可以直接使用函数,会使代码更简单

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
if (nums.empty()) {
return {-1, -1};
}
int lower = lower_bound(nums.begin(), nums.end(), target) - nums.begin();
int upper = upper_bound(nums.begin(), nums.end(), target) - nums.begin() - 1;
if (upper < 0 || lower == nums.size() || nums[upper] != target) {
return {-1, -1};
}
else
return {lower, upper};
}
};