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

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

如果数组中不存在目标值 target，返回 [-1, -1]。

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

示例 1：

输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]

示例 2：

输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]

示例 3：

输入：nums = [], target = 0
输出：[-1,-1]

解题思路：

为了满足时间复杂度 的要求，我们可以使用二分查找算法。二分查找的基本思想是每次将搜索区间缩小一半，从而快速定位目标元素。由于要找到目标值的开始位置和结束位置，我们需要进行两次二分查找，一次用于找到目标值的起始位置，另一次用于找到目标值的结束位置。

查找起始位置的函数 findLeft

初始化指针：设置左指针 left 指向数组的起始位置（索引为 0），右指针 right 指向数组的结束位置（索引为 nums.length - 1），同时初始化结果变量 result 为 -1，表示初始时未找到目标值。

• 二分查找过程：
  • 当 left 小于等于 right 时，进行循环。
  • 计算中间位置 mid，为了避免整数溢出，使用 mid = left + (right - left) / 2 进行计算。
  • 如果 nums[mid] 大于等于 target，说明目标值可能在左半部分，更新 right = mid - 1。如果此时 nums[mid] 恰好等于 target，则将 result 更新为 mid，因为我们要找的是起始位置，所以继续向左搜索。
  • 如果 nums[mid] 小于 target，说明目标值在右半部分，更新 left = mid + 1。
• 返回结果：循环结束后，返回 result，如果未找到目标值，result 仍为 -1。

查找结束位置的函数 findRight

初始化指针：同样设置左指针 left 指向数组的起始位置，右指针 right 指向数组的结束位置，结果变量 result 初始化为 -1。

• 二分查找过程：
  • 当 left 小于等于 right 时，进行循环。
  • 计算中间位置 mid。
  • 如果 nums[mid] 小于等于 target，说明目标值可能在右半部分，更新 left = mid + 1。如果此时 nums[mid] 等于 target，则将 result 更新为 mid，因为我们要找的是结束位置，所以继续向右搜索。
  • 如果 nums[mid] 大于 target，说明目标值在左半部分，更新 right = mid - 1。
  • 返回结果：循环结束后，返回 result。

具体代码:

import java.util.Scanner;
import java.util.Arrays;

// 解决方案类，用于实现查找目标值在数组中的起始和结束位置的功能
class Solution34 {
    // 主方法，调用 findLeft 和 findRight 方法分别找到目标值的起始和结束位置
    public int[] searchRange(int[] nums, int target) {
        int leftIndex = findLeft(nums, target);
        int rightIndex = findRight(nums, target);
        return new int[]{leftIndex, rightIndex};
    }

    // 二分查找目标值的起始位置
    private int findLeft(int[] nums, int target) {
        // 初始化左指针为数组起始位置
        int left = 0;
        // 初始化右指针为数组结束位置
        int right = nums.length - 1;
        // 用于记录目标值的起始位置，初始化为 -1 表示未找到
        int result = -1;

        // 当左指针小于等于右指针时，继续二分查找
        while (left <= right) {
            // 计算中间位置，避免整数溢出
            int mid = left + (right - left) / 2;

            if (nums[mid] >= target) {
                // 如果中间元素大于等于目标值，更新右指针到中间位置的前一个位置
                right = mid - 1;
                if (nums[mid] == target) {
                    // 如果中间元素等于目标值，记录当前位置为可能的起始位置
                    result = mid;
                }
            } else {
                // 如果中间元素小于目标值，更新左指针到中间位置的后一个位置
                left = mid + 1;
            }
        }
        return result;
    }

    // 二分查找目标值的结束位置
    private int findRight(int[] nums, int target) {
        // 初始化左指针为数组起始位置
        int left = 0;
        // 初始化右指针为数组结束位置
        int right = nums.length - 1;
        // 用于记录目标值的结束位置，初始化为 -1 表示未找到
        int result = -1;

        // 当左指针小于等于右指针时，继续二分查找
        while (left <= right) {
            // 计算中间位置，避免整数溢出
            int mid = left + (right - left) / 2;

            if (nums[mid] <= target) {
                // 如果中间元素小于等于目标值，更新左指针到中间位置的后一个位置
                left = mid + 1;
                if (nums[mid] == target) {
                    // 如果中间元素等于目标值，记录当前位置为可能的结束位置
                    result = mid;
                }
            } else {
                // 如果中间元素大于目标值，更新右指针到中间位置的前一个位置
                right = mid - 1;
            }
        }
        return result;
    }
}

// 测试类，用于测试 Solution34 类的功能
public class TestSolution {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        // 提示用户输入数组元素，元素之间用逗号分隔
        System.out.println("请输入一个非递减顺序排列的整数数组，元素之间用逗号分隔：");
        String input = scanner.nextLine();
        // 将输入的字符串按逗号分割成字符串数组
        String[] numStrings = input.split(",");
        // 创建整数数组来存储分割后的元素
        int[] nums = new int[numStrings.length];
        for (int i = 0; i < numStrings.length; i++) {
            // 将字符串元素转换为整数并存储到数组中
            nums[i] = Integer.parseInt(numStrings[i].trim());
        }

        // 提示用户输入目标值
        System.out.println("请输入要查找的目标值：");
        int target = scanner.nextInt();

        // 创建 Solution34 类的实例
        Solution34 solution = new Solution34();
        // 调用 searchRange 方法查找目标值的起始和结束位置
        int[] result = solution.searchRange(nums, target);

        // 输出结果
        System.out.println("目标值 " + target + " 在数组中的起始和结束位置是：" + Arrays.toString(result));

        scanner.close();
    }
}

运行截图：