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
知识点
二分查找
Java代码
public static int searchInsert(int[] nums, int target) {
return bs2(nums, target, 0, nums.length - 1);
}
//二分查找 递归
public static int bs(int[] nums, int target, int q, int p) {
if(q > p){
return q;
}
int mid = (q + p) >> 1;
if (nums[mid] > target){
return bs(nums, target, q, mid - 1);
} else if(nums[mid] < target){
return bs(nums, target, mid + 1, p);
} else {
return mid;
}
}
//二分查找 循环
public static int bs2(int[] nums, int target, int q, int p) {
int mid;
while(q <= p){
mid = (q + p) >> 1;
if (nums[mid] > target){
p = mid - 1;
} else if(nums[mid] < target){
q = mid + 1;
} else {
return mid;
}
}
return q;
}