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

https://leetcode.cn/problems/find-first-and-last-position-of-element-in-sorted-array/description/

1.题目描述

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

2.解题思路

给定的nums数组是非递减排列，每个数可以有多个，求target的范围。

此时用传统的二分查找已经不行了，因为传统的二分查找找的是某一个数，而这道题找的是某一个数的范围。此时可以利用左右二分查找解决此题

1.可以以 target 为界，利用二分查找找出 target 的左侧边界；然后利用二分查找找到 target 的右边界

2.利用二分查找找左边界

• 定义 left = 0；right = nums.length；mid = left + (right - left)/2

• 当 nums[mid] < target 时，left 要改变，left = mid + 1

• 当 nums[mid] > target 时， right 要改变，right = mid;

• 当 left < right 时，重复以上三步，直到 left==right，

• 判断nums[mid] 是否等于 target，如果相等，此时 mid 就是 target 的左边界，将mid存入index数组；否则将 -1 存入index数组

3.利用二分查找找右边界

• 重新赋值 left 和 right，left = 0；right = nums.length；此时mid要改变：mid = left + (right - left + 1) / 2

• 当 nums[mid] <= target，left = mid

• 当 nums[mid] > target，right = mid-1

• 重复上述三步直到 left == right

• 判断nums[mid] 是否等于 target，如果相等，此时 mid 就是 target 的左边界，将mid存入index数组；否则将 -1 存入index数组

4.返回index数组即可

特殊情况：

当nums为空时，需要特殊处理，将index中放入[-1,-1]，然后返回即可

3.代码

public static int[] searchRange(int[] nums, int target) {
    int[] index = new int[2];
    int left = 0;
    int right = nums.length-1;

    if (nums.length == 0){
        index[0] = -1;
        index[1] = -1;
        return index;
    }
    //找target左边界
    while (left < right){
        int mid = left+(right-left)/2;
        if (nums[mid] < target){
            left = mid+1;
        }else {
            right = mid;
        }
    }
    //right == left 时
    if (nums[left] == target){
        index[0] = left;
    }else {
        index[0] = -1;
        index[1] = -1;
        return index;
    }

    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;
        }
    }
    if (nums[left] == target){
        index[1] = left;
    }
    return index;

}