[路飞]_leetcode刷题_35. 搜索插入位置

212 阅读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

示例 4:

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

示例 5:

输入: nums = [1], target = 0
输出: 0

思路:

这个可以使用二分法来解。

经典的二分法题目#### 704. 二分查找就是在一个有序数列里找目标值,找到则返回,没有找到则返回-1;

那我们要分析出这一题和经典题有什么不一样。

我们这里其实是要找到target的位置 ,target满足 nums[ans-1] < target <=nums[ans]

如果刚好找到,则返回ans;如果找不到,那就往ans插入,也是返回ans

所以可以理解为我们要找的位置,是等于target的位置,或者第一个大于target的位置

二分查找,最终left和right会指向同一个数,即left=right,

如果mid>left,则left+1;如果mid<left,则right-1

可以看到不管是那种情况,最终left所在的位置,就是第一个大于target的位置

所以如果没有找到mid=target,跳出循环后,return left即可

如果循环中找到了相等的数,则直接return mid即可

可以看到,这704.二分查找相比,就是return -1变为了return left

代码如下:

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number}
 */
var searchInsert = function(nums, target) {
    let n = nums.length;   
    let left = 0,right = n-1;

    while(left <=right){
        let mid = Math.floor((left+right)/2);
        if(nums[mid] == target){
            return mid
        }else if(nums[mid] > target){
            right = mid - 1
        }else{
            left = mid +1
        }
    }
    return left
};