LeetCode34 在排序数组中查找元素

发表于 | 分类于 |
字数统计: 306 | 阅读时长 ≈ 1

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

题面

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

这道题目可以说是直接告诉你要用二分来做，那么具体来说就是会用到C++里的lower_bound和upper_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
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;
    }

};

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

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};
    }
};