每天一道LeetCode-搜索插入位置

86 阅读1分钟

第10道算法题

题目要求:给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

来源:力扣(LeetCode) 链接:leetcode.cn/problems/se…

我的思路:整体思路使用二分法,先将目标值在数组两端外的返回对应位置,之后用二分法逐步缩小目标范围,最后得出的值一定在最终范围内的闭区间的两端之一,再根据是否与中间值相等进行判断

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number}
 */
var searchInsert = function(nums, target) {
    if(nums[nums.length-1]<target)return nums.length
    if(nums[0]>target)return 0

    let l = 0
    let r = nums.length-1 
    let m = Math.floor((nums.length-1)/2)
    while(r>=l){
        if(nums[m]===target) return m

        if(r-l===1){
            return r
        }

        if(nums[m]<target){
            l = m
            m = Math.floor((r+l)/2)
        }
        if(nums[m]>target){
            r= m
            m = Math.floor((r+l)/2)
        }
    }
};