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

题目描述

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

示例 ：

输入：nums = [5,7,7,8,8,8,9,10], target = 8

输出：[3,5] 

解题思路

在解决该题是我们可以发现，如果我们还采取与此前(上一篇博客中)相同的方法，找出目标元素后，因为无法确定其是否该元素的第一个或最后一个位置，所以我们还需要再次向前以及向后遍历来找到结果。但是这样无疑会降低效率，所以我们需要对其进行一定优化。

查找区间左端点：

nums[mid] < target : left =mid + 1;

nums[mid] >= target : right = mid;

细节处理

循环条件：

left < right

left = right 时就是最终结果，无需再继续判断，否则会进入死循环。

求中点操作：

left + (right - left ) /2 ,而非 left + (right - left + 1 ) / 2 。

因为倘若在只剩下两个元素后，前一种方法取得的mid 是left，后一种方法取得的是right，如果取得right，在nums[mid] >= target 的情况下，right = mid，将会出现死循环。

查找区间右端点：

nums[mid] > target : right =mid - 1;

nums[mid] <= target : left = mid;

循环条件：

left < right

left = right 时就是最终结果，无需再继续判断，否则会进入死循环。

求中点操作：

left + (right - left + 1 ) /2 ,而非 left + (right - left ) / 2 ，原因与上相近。

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> ret;
        if (nums.empty()) {
            ret.push_back(-1);
            ret.push_back(-1);
            return ret;
        }
        int left = 0, right = nums.size() - 1, mid;
        while (left < right) {
            mid = left + (right - left) / 2;
            if (nums[mid] < target) {
                left = mid + 1;
            } else {
                right = mid;
            }
        }
        if (nums[left] != target) {
            ret.push_back(-1);
            ret.push_back(-1);
            return ret;
        }
        ret.push_back(left);
        right = nums.size() - 1;
        while (left < right) {
            mid = left + (right - left + 1) / 2;
            if (nums[mid] > target) {
                right = mid - 1;
            } else {
                left = mid;
            }
        }        
        ret.push_back(right);
        return ret;
    }
};

代码逻辑

1. 初始化：
   • 创建一个空的结果向量 ret。
   • 如果 nums 为空，则将 -1 两次推入 ret 并返回。
2. 查找开始位置：
   • 使用 left 和 right 指针（索引）在数组中进行二分查找。
   • 当 left < right 时，循环继续。
   • 在循环中，计算中间索引 mid，并根据 nums[mid] 与 target 的比较结果更新 left 或 right。
   • 重要的是，当 nums[mid] >= target 时，应将 right 更新为 mid 而不是 mid - 1，因为 mid 可能就是目标值的开始位置。
   • 循环结束后，检查 nums[left] 是否等于 target。如果不等，则返回 [-1, -1]；否则，将 left（即开始位置）推入 ret。
3. 查找结束位置：
   • 重新将 right 设置为数组的最后一个索引。
   • 再次使用二分查找的逻辑，但这次是为了找到目标值的结束位置。
   • 需要注意的是，在更新 mid 时，使用了 mid = left + (right - left + 1) / 2;，这是为了避免在 left 和 right 相邻时陷入无限循环，并确保当 left 和 right 指向同一个元素时，mid 会等于 right（即当前考虑的最后一个可能的位置）。
   • 如果 nums[mid] 大于 target，则将 right 更新为 mid - 1；否则，将 left 更新为 mid。
   • 当 left == right 时，循环结束，此时 right（或 left，它们现在相同）就是目标值的结束位置。
   • 将结束位置推入 ret。