今天遇到一个问题:给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
栗子:对于给定排序数组 [1,3,5] 和 目标值 4,如果数组中能找到4就返回其下标,如果不能就插入对应排序位置(也就是3后面),且返回对应下标,且题目要求时间复杂度为 O(log n) 的算法,那就是用二分法了
1、首先解决查找下标问题
function getIndex(arr,num){
let len = arr.length,
st = 0,
mid = 0,
end = len-1;
// 二分法其实就是不断择中进行对比
while(st <= end){
mid = Math.floor((st + end) /2)
if(num == arr[mid]){
return mid
}else if(num > arr[mid]){
st = mid + 1
}else{
end = mid - 1
}
}
return mid
}
2、根据下标值,判断与目标值大小
let res = getIndex(nums, target),
index = res,
now = nums[index];
// 判断当前下标值与目标值大小
if(now > target) {
// 利用splice插入指定位置,且修改原数组
nums.splice(index,0,target)
}else if(target > now){
res = index + 1
nums.splice(res,0,target)
}
return res