「优选算法刷题」:在排序数组中查找元素的第一个和最后个位置

一、题目

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

二、思路解析

二分查找，它很简单，但也很容易写出死循环。不过，不必过多恐惧，只要多做练习，他就会是最简单的查找算法！

我们来看这道题，主要分为 2 部分：查找区间的左端点 和 右端点。

1）查找区间左端点

左边界划分的两个区间的特点：

▪ 左边区间 [left, resLeft - 1] 都是⼩于 x 的；
▪ 右边区间（包括左边界） [resLeft, right] 都是⼤于等于 x 的；

因此，关于 mid 的落点，我们可以分为下⾯两种情况：

◦ 当我们的 mid 落在 [left, resLeft - 1] 区间的时候，也就是 arr[mid] <target 。说明 [left, mid] 都是可以舍去的，此时更新 left 到 mid + 1 的位置，继续在 [mid + 1, right] 上寻找左边界；

◦ 当 mid 落在 [resLeft， right] 的区间的时候，也就是 arr[mid] >= target 。说明 [mid + 1, right] （因为 mid 可能是最终结果，不能舍去）是可以舍去的，此时更新 right 到 mid 的位置，继续在 [left, mid] 上寻找左边界；

注意：这⾥找中间元素需要向下取整，即 mid = left + ( right - left ) / 2 ，而不是 mid = left + ( right - left + 1 ) / 2 。

因为后续移动左右指针的时候：
• 左指针： left = mid + 1 ，是会向后移动的，因此区间是会缩⼩的；
• 右指针： right = mid ，可能会原地踏步（⽐如：如果向上取整的话，如果剩下 1,2 两个元
素， left == 1 ， right == 2 ， mid == 2 。更新区间之后， left，right，mid 的
值没有改变，就会陷⼊死循环）。
因此⼀定要注意，当 right = mid 的时候，要向下取整。

2）查找区间右端点

我们先⽤ resRight 表⽰右边界；

这时可以注意到右边界的特点：

        ▪ 左边区间 （包括右边界） [left, resRight] 都是⼩于等于 x 的；
        ▪ 右边区间 [resRight+ 1, right] 都是⼤于 x 的；

因此，关于 mid 的落点，我们可以分为下⾯两种情况：

◦ 当我们的 mid 落在 [left, resRight] 区间的时候，说明 [left, mid - 1]（ mid 不可以舍去，因为有可能是最终结果） 都是可以舍去的，此时更新 left 到 mid的位置；
◦ 当 mid 落在 [resRight+ 1, right] 的区间的时候，说明 [mid, right] 内的元素是可以舍去的，此时更新 right 到 mid - 1 的位置；

• 由此，就可以通过⼆分，来快速寻找右边界；
注意：这⾥找中间元素需要向上取整「 mid = left + ( right - left + 1 ) / 2」。

因为后续移动左右指针的时候：
• 左指针： left = mid ，可能会原地踏步（比如：如果向下取整的话，如果剩下 1,2 两个元
素， left == 1， right == 2，mid == 1 。更新区间之后， left，right，mid ?的值
没有改变，就会陷⼊死循环）。
• 右指针： right = mid - 1 ，是会向前移动的，因此区间是会缩小的；
因此⼀定要注意，当? right = mid ?的时候，要向下取整。

三、完整代码

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int ret[] = new int[2];
        ret[0] = ret[1] = -1;

        // 处理边界情况
        if(nums.length == 0){
            return ret;
            }

        // 1. ⼆分左端点    
        int left = 0;
        int right = nums.length - 1;
        while(left < right){
            int mid = left + (right - left) / 2;
            if(nums[mid] < target){
                left = mid + 1; 
            }else{
                right = mid;
            }
        }

        // 判断是否有结果
        if(nums[left] != target){
            return ret;
        }else{
            ret[0] = left;
        }

        // 2. ⼆分右端点
        left = 0;
        right = nums.length - 1;
        while(left < right){
            int mid = left + (right - left + 1) / 2;
            if(nums[mid] <= target){
                left = mid;
            }else{
                right = mid - 1;
            }
        }
        ret[1] = right;
        return ret ;
    }
}

以上就是本篇博客的全部内容啦，如有不足之处，还请各位指出，期待能和各位一起进步！