解决数组中目标值范围问题

问题描述

我们需要找到一个数组 nums 中的目标值 target 的范围，即找到 target 的第一个和最后一个出现位置。如果 target 不在数组中，返回一个空数组。

解题思路

我们可以定义一个辅助方法 extremeInsertionIndex，用于查找目标值 target 在数组中的开始位置或结束位置。这个方法有两个参数：

left 参数为 true 时，返回目标值的开始位置；

false 时返回目标值的结束位置。

查找开始位置的逻辑

初始化边界：

左边界 lo 设为 0；

右边界 hi 设为数组长度。

使用二分查找法：

计算中间值 mid；

如果 nums[mid] 大于目标值，说明目标值在左边界，更新右边界 hi 为 mid；

如果 nums[mid] 等于目标值，可能 mid 就是开始位置，更新右边界 hi 为 mid；

当边界闭合时，返回左边界 lo。

查找结束位置的逻辑

同样初始化边界 lo 和 hi；

如果 nums[mid] 大于目标值，更新右边界 hi 为 mid；

如果 nums[mid] 等于目标值，更新左边界 lo 为 mid + 1；

返回右边界 hi 并减 1。

解题代码

class Solution {    private int extremeInsertionIndex(int[] nums, int target, boolean left) {        int lo = 0, hi = nums.length;        while (lo < hi) {            int mid = lo + (hi - lo) / 2;            if (nums[mid] > target || (left && target == nums[mid])) {                hi = mid;            } else {                lo = mid + 1;            }        }        return lo;    }    public int[] searchRange(int[] nums, int target) {        int[] result = {-1, -1};        int leftIdx = extremeInsertionIndex(nums, target, true);        if (leftIdx == nums.length || nums[leftIdx] != target) {            return result;        }        result[0] = leftIdx;        result[1] = extremeInsertionIndex(nums, target, false) - 1;        return result;    }}