L-4: 34--在排序数组中查找元素的第一个和最后一个位置

发布时间 2023-11-06 15:02:49作者: 翻斗花园小美Q

给你一个按照非递减顺序排列的整数数组 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]
class Solution {
    public int[] searchRange(int[] nums, int target) {
        int first = -1, last = -1; 

        //两次二分查找,分别找第一个和第二个
        //思路:找到mid = target之后,然后分两部分直接用mid左右挪,找到第一个位置和最后一个位置
        //left和right在下面还需要重新初始化

        int left = 0, right = nums.length - 1;
        while (left <= right){
            int mid = (left + right) / 2;
            if (target < nums[mid]){
                right = mid - 1;
            }else if (target > nums[mid]){
                left = mid + 1;
            }else {
                //找第一个最左第一个等于target的位置
                first = mid;
                right = mid - 1;//重要,right左移,这句可以实现找到最左边第一次出现的target(先不管半右部分)
            }
        }

            //第二次查找最后一个等于target的位置
            //这次需要查找右半部分的,所以 left右移 ;left = mid + 1
            left = 0; right = nums.length - 1;
            while (left <= right){
            int mid = (left + right) / 2;
            if (target < nums[mid]){
                right = mid - 1;
            }else if (target > nums[mid]){
                left = mid + 1;
            }else {
                //找最右等于target的位置
                last = mid;
                left = mid + 1;//重要,right左移,这句可以实现找到最左边第一次出现的target(先不管半右部分)
                 }
            }
        return new int[] {first, last};
    }
}