leetcode34.在排序数组中查找元素的第一个和最后一个位置

Posted on | In |
Words count in article: 312 | Reading time ≈ 1

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

示例：

1
2
输入: nums = [1,3,5,6], target = 5
输出: 2

分析：

1
2
3
两次二分：
	- 从左到右，找右边界
	- 从右到左，找左边界

代码

  • C++

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
class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        if (nums.size() == 0) return {-1, -1};
        int l = 0, r = nums.size() - 1;
        vector<int> res(2, -1);
        while (l < r) {
            int mid = l + r + 1 >> 1;
            if (nums[mid] <= target) l = mid;
            else r = mid - 1;
        }
        if (nums[l] == target)
            res[1] = l;

        l = 0, r = nums.size() - 1;
        while (l < r) {
            int mid = l + r >> 1;
            if (nums[mid] >= target) r = mid;
            else l = mid + 1;
        }
        if (nums[l] == target)
            res[0] = r;

        return res;
    }
};class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int l = 0, r = nums.size() - 1;
        while (l < r) {
            int mid = l + r >> 1;
            if (nums[mid] >= target) r = mid;
            else l = mid + 1;
        }

        return target > nums.back() ? nums.size() : l; 
    }
};

[原题链接](34. 在排序数组中查找元素的第一个和最后一个位置 - 力扣（Leetcode）)

0%