给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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
【分析】:
本题是一道非常适合用二分查找解决的问题。
注意一个点:
当low和high相等的时候有三种情况。具体看代码
【代码】:
/**
* @param {number[]} nums
* @param {number} target
* @return {number}
*/
var searchInsert = function(nums, target) {
let low = 0;
let high = nums.length - 1;
let mid;
while(low < high){
mid = parseInt((high + low) / 2);
if(nums[mid] == target){
return mid;
}else if(nums[mid] < target){
low = mid + 1;
}else{
high = mid - 1;
}
}
//low与high相等的时候
if(nums[low] == target){
return low;
}else if(nums[low] < target){ //比如[1,3,5,6] target=7
return low + 1;
}else{
//比如[1,2,3,4] target=0
return low;
}
};