leetCode 34. 在排序数组中查找元素的第一个和最后一个位置-CSDN博客

本文链接：https://blog.csdn.net/weixin_42324665/article/details/125618967

给你一个按照非递减顺序排列的整数数组 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的时候，就满足了条件，返回当前下标,循环结束没有满足，返回-1

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
 var searchRange = function(nums, target) {
    let left = 0, right = nums.length-1
    while(left <= right){
        if(nums[left] < target){
            left++
        }
        if(nums[right] > target){
            right--
        }
        if(nums[left] === target && nums[right] === target){
            return [ left , right]
        }
    }

    return [-1,-1]
};

二、二分查找

leftIdx 为第一个满足条件的下标，rightIdx 数组中第一个大于target 的下标，然后将下标加一，两个的判断条件不同，为了将代码复用，封装一个binarySearch(nums,target,flag) 函数

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
 var searchRange = function(nums, target) {
    function binarySearch(nums,target,flag){

        let left = 0, right = nums.length-1,ans = nums.length
        while(left <= right){
            let mid = Math.floor((right + left) / 2) 
            if(nums[mid] > target || (flag && nums[mid] >= target)){
                right = mid -1 
                ans = mid
            }else{
                left = mid + 1
            }
        }
        return ans
    }
    let ans = [-1,-1]
    let leftIdx = binarySearch(nums,target,true) // 找出第一个出现的位置
    let rightIdx = binarySearch(nums,target,false)-1 // 找对第二个出现的位置的下一个
    if(leftIdx <= rightIdx && rightIdx < nums.length && nums[leftIdx] === target && nums[rightIdx] === target){
        ans = [leftIdx,rightIdx]
    }
    return ans

};