题目
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
};