算法题-14 - 技术栈

算法题-14

白中白121382026-02-14 12:59

给你一个按照非递减顺序排列的整数数组 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的边界，那么我们可以使用两次二分查找分别去寻找target的左右边界，找左边界nums[mid] >= target，我们就往左缩：right = mid - 1，找右边界nums[mid] <= target，我们往右扩：left = mid + 1，具体代码简单

javascript 复制代码

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
var searchRange = function(nums, target) {
    const findLeft = () => {
        let left = 0
        let right = nums.length - 1
        let res = -1

        while (left <= right) {
            let mid = Math.floor((left + right) / 2)

            if (nums[mid] >= target) {
                right = mid - 1
            } else {
                left = mid + 1
            }

            if (nums[mid] === target) {
                res = mid
            }
        }

        return res
    }

    const findRight = () => {
        let left = 0
        let right = nums.length - 1
        let res = -1

        while(left <= right){
            let mid = Math.floor((left + right) / 2)

            if (nums[mid] <= target) {
                left = mid + 1
            
            } else {
                right = mid - 1
            }

            if (nums[mid] === target) {
                res = mid
            }
        }

        return res
    }
    return [findLeft(), findRight()]
};