力扣LeetCode #34 在排序数组中查找元素的第一个和最后一个位置（SearchRange）

最新推荐文章于 2021-12-11 19:13:57 发布

苏志林的情敌

最新推荐文章于 2021-12-11 19:13:57 发布

阅读量114

点赞数

文章标签：算法数据结构 leetcode java 二分查找

本文链接：https://blog.csdn.net/weixin_49155948/article/details/111615352

版权

- 题目描述

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。如果数组中不存在目标值 target，返回 [-1, -1]。
【进阶】你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗？

来源：LeetCode

- 示例

示例 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]

- 思路分析

这道题有点类似#33，都是在二分查找的基础上变化一点。当二分查找的 $n u m s [m i d] = t a r g e t$ 时，有三种情况：

$n u m s [m i d + 1] \neq = t a r g e t$ ：此时说明找到的 $n u m s [m i d]$ 就是目标数的结束位置。需要注意的是，因为可能只有一个数，因此当我们第一次找到开始位置或者结束位置时，我们就将开始位置和结束位置都设置为找到的这个 $m i d$ 。
$n u m s [m i d - 1] \neq = t a r g e t$ ：此时找到的 $n u m s [m i d]$ 是目标数的开始位置。如果是第一次找到，同样将开始位置和结束位置都设置为找到的这个 $m i d$ 。
$n u m s [m i d - 1] = n u m s [m i d + 1]$ ：说明找到的是目标数的中间位置。此时则分别二分查找前半段数组和后半段数组即可。

需要特别注意的是边界判断， $m i d - 1$ 和 $m i d + 1$ 都有可能导致数组越界。

- JAVA实现

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int len = nums.length;
        int[] ans = {-1,-1};  //将结果数组初始化为-1
        if(len==0) return ans;
        int start = 0, end = len-1;
        abs(nums, target, start, end, ans);
        return ans;
    }

    public static void abs(int[] nums, int target, int start, int end, int[] ans) {
        if(start<=end) {
            int mid = (start+end)/2;
            if(nums[mid]==target) {
                if (mid-1<0 || nums[mid - 1] != target) {  //找到是开始位置
                    ans[0] = mid;
                    if (ans[1] == -1) {  //如果结尾位置还没有找到
                        ans[1] = mid;
                        abs(nums, target, mid + 1, end, ans);   //二分查找后半段找结尾位置
                    }
                    else return;
                }
                else if (mid+1>=nums.length || nums[mid + 1] != target) { //找到的是结束位置
                    ans[1] = mid;
                    if (ans[0] == -1) {   //如果开始位置还没有找到过
                        ans[0] = mid;
                        abs(nums, target, start, mid - 1, ans); //二分查找前半段找开始位置
                    }
                    else return;
                }
                else { //找到的是中间位置
                    abs(nums, target, start, mid - 1, ans);  
                    abs(nums, target, mid + 1, end, ans);
                }
            }
            else if(nums[mid]<target) abs(nums, target, mid+1, end,ans);
            else abs(nums, target, start, mid-1, ans);
        }
    }
}