21-06-05-搜索插入位置

题目地址

https://leetcode-cn.com/problems/search-insert-position/

题目描述

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

你可以假设数组中无重复元素。

示例 1:

输入: [1,3,5,6], 5
输出: 2
示例 2:

输入: [1,3,5,6], 2
输出: 1
示例 3:

输入: [1,3,5,6], 7
输出: 4
示例 4:

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

思路

• 排序，查找，这两个关键词就要想到二分查找。
• 只不过题目稍作变种。理解下来，其实就是找到第一个大于等于 target 的地方。
• 等于的时候就是跟二分无差别。
• 要返回插入位置，其实就是第一个大于 target的位置就是插入的地方。

代码

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

复杂度分析

• 时间复杂度：O(logN)
• 空间复杂度：O(1)