LeetBook乐扣题库 34. 在排序数组中查找元素的第一个和最后一个位置

一 题目描述

给你一个按照非递减顺序排列的整数数组 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]

二 方法代码

• 方法: 二分查找

通过二分查找得到target的边界，即左边界为第一个target的上一位，右边界为最后一个target的下一位，通过使用二分查找的左右指针来找边界，正常二分查找当mid的值等于target时，会直接return，返回时可能左指针 left 仍小于右指针 right，但是本题需要注意，当mid 的值等于target的时候，继续执行while循环，直到left和right相等。

首先，左右边界的初始值设为-2的原因：因为可能左右边界的值最后会是-1，比如【2，2】的时候，target为3时左边界为-1，所以初始值不能设置为-1

找左边界，要通过右指针来找，因为为了寻找左边界，当mid的值等于target时，要进一步探索左边的区域（因为不确定这是否是第一个target，所以继续探索左边），因此要修改right，即right = mid - 1；并把right的值赋给左边界的变量。left是为了压缩左边的区域的空间，更快地找到左边界

    public int getLeft(int[] nums, int target){
        int left = 0, right = nums.length - 1;
        //可能左右边界的值最后会是-1，比如【2，2】的时候，target为3时左边界为-1
        int leftBound = -2;
        while(left <= right){
            int mid = left + (right - left) / 2;
            if(nums[mid] >= target){
                //继续探索左边的区域，找到第一个target的前面位置
                right = mid - 1;
                leftBound = right;
            }else{
                left = mid + 1;
            }
        }
        return leftBound;
    }

右边界的情况是类似的，用左指针

    public int getRight(int[] nums, int target){
        int left = 0, right = nums.length - 1;
        int rightBound = -2;
        while(left <= right){
            int mid = left + (right - left) / 2;
            if(nums[mid] <= target){
                left = mid + 1;
                rightBound = left;
            }else{
                right = mid - 1;
            }
        }
        return rightBound;
    }

本题有三种情况：

• 情况一：target 在数组范围的右边或者左边，例如数组{3, 4, 5}，target为2或者数组{3, 4, 5},target为6，此时应该返回{-1, -1}
• 情况二（需要注意，见下面黄色）：target 在数组范围中，且数组中不存在target，例如数组{3,6,7},target为5，此时应该返回{-1, -1}
• 情况三：target 在数组范围中，且数组中存在target，例如数组{3,6,7},target为6，此时应该返回{1, 1}

情况二需要注意，比如题目的例子2，target为6，nums = [5,7,7,8,8,10]，此时左右边界就是0和1，即nums[0]=5和nums[1]=7，这种情况就是左右边界的差值为1，而正常情况下左右边界的差值会至少为2，比如说target = 5，此时左右边界-1和1，差值为2，这会用于后面的 if 判断

    public int[] searchRange1(int[] nums, int target) {
        int leftBound = getLeft(nums, target);
        int rightBound = getRight(nums, target);

        //情况1
        if(leftBound == -2 || rightBound == -2){
            return new int[]{-1, -1};
        }
        //情况3
        else if(rightBound - leftBound > 1){
            return new int[]{leftBound+1, rightBound-1};
        }
        //情况2
        else{
            return new int[]{-1, -1};
        }
    }

三 运行结果