算法题-数组-35-搜索插入位置

75 阅读1分钟

35-搜素插入位置

题目描述

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

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

示例 1:

输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:

输入: nums = [1,3,5,6], target = 2
输出: 1

示例 3:

输入: nums = [1,3,5,6], target = 7
输出: 4

分析

题目提示是 O(log n) 可以考虑使用二分查找,因为二分查找就是找到目标元素的位置,如果找不到,则当前的 left 和 right 应该相等,如果我们的区间左右都是闭区间,那么 right+1 即是找不到的话的插入位置。

题解

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

分析

时间复杂度:O(log n)

空间复杂度:O(1)

难点

区间划分,边界条件