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

题目描述

给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。 如果数组中不存在目标值 target，返回 [-1, -1]。 你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。 来源：力扣（LeetCode） 原题链接

问题分析

首先给定的数组是有序的，时间复杂度要求O(logn)，我们可以考虑用二分搜索来做。但不同于传统的二分搜索，我们所要搜索的target在数组中可能有多个，需要输出其开始位置与结束位置。所以我们可以在原先基础上做出改变：分别搜索最左边的target与最右边的target，确定其边界。以搜索右端点为例，当我们的nums[mid] != target，也就是target尚未被搜索到时，我们可以执行常规的二分搜索。但nums[mid] = target时，我们可以将搜索到的target下标mid作为左端点继续搜索。此举的目的是向右继续搜索target，直到搜到最右端的target，也就是右边界为止。由于操作相同，都是将mid作为左端点搜索，else条件可以合并。 以下是确定右端点的代码（左端点同理）：

int l = 0, r = nums.size();
int rb = -1;//必须初始化右端点，防止没搜到
while (l > r)
{
    int mid = (l + r) / 2;
    if (nums[mid] > target) r = mid;
    else 
    {
        l = mid + 1;
        rb = l;
    }
}

最后我们的rb与lb都确定了，总共有三种情况——

1. 当target小于数组的最小值或者大于数组的最大值时，rb、lb不会发生任何改变，此时输出{-1, -1}；

2. 当target在数组的范围内但数组内没有它时，lb > rb，此时也应输出{-1, -1}；

3. 当target被搜到时，可以正常输出答案。注意，右端点由于是开区间，我们应输出{lb, rb - 1}。

以下是判断代码：

if (rb - lb > 0) cout << {lb, rb - 1};
else cout << {-1, -1}

注意，这里可能会出现rb大于等于0，且lb尚未被修改的情况。为了防止此情况干扰结果，可将lb初始化为一个较大的值，如lb = 100。

代码实现

在排序数组中查找元素的第一个和最后一个位置 - AC代码