LeetCode：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

提示:

1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 为 无重复元素 的 升序 排列数组
-104 <= target <= 104

解题思路

首先分析题目，这是一个有序的数组，查找指定值，满足二分的基本条件

这里是一个升序的数组，如果说num[mid] >= target,那么最终答案会落在左半区间，因为这又是一个升序的数组，如果说mid指向的元素都比目标值大了，那么后面的只可能更大，可以尝试使用整数二分模板一。

最终两个指针都会指向同一个位置，可以通过判断最后指向的位置是否为目标值，来判断是否有解，即使目标值不存在于数组中，只要目标值的大小在数组的最小值与最大值之间，也会返回目标插入的下标

当目标值大于数组的最大值时，直接返回数组的长度，即为目标值在数组中的下标

JavaScript 代码实现

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

C++ 代码实现

class Solution
{
public:
  int searchInsert(vector<int> &nums, int target)
  {
    int l = 0, r = nums.size() - 1, max = nums.size();
    while (l < r)
    {
      int mid = l + r >> 1;
      if (nums[mid] >= target) r = mid;
      else l = mid + 1;
    }

    if (nums[l] < target) return max;
    else return l;
  }
};

Java 代码实现

class Solution {
    public int searchInsert(int[] nums, int target) {
      int l = 0, r = nums.length - 1, max = nums.length;
      while (l < r){
        int mid = l + r >> 1;
        if (nums[mid] >= target) r = mid;
        else l = mid + 1;
      }

      if (target > nums[r]) return max;
      else return l;
    }
}