题目描述:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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
思路:
这道题依然是基于二分法查找有序数组的进阶,目的是训练我们在有限条件下查找数的位置,题目的前半部分,若存在数则返回下标,是基础二分法的题解,后半部分为不存在的情况则返回「应该」插入的位置,我们需要在原二分法的基础上做出修改,其中一种思路是,始终使二分法窗口的右边界+1为插入目标,因为若左边界为0,则插入目标不能为数组下标-1的位置。
误区:
注意解析题目,我们寻找的是应该插入的位置,所以不用进行插入操作,误解了会浪费很多时间去做数组插入操作,其次是我们不能想太复杂而进入过多的if-else语句,应该在二分法的基础上再优化
题解:
public int searchInsert(int[] nums, int target) {
//取到数组的长度
int n = nums.length;
//使用二分法创建窗口
int left = 0, right = n - 1, ans = n;
while (left <= right) {
//位运算 >> 等价于 /2,依然是常见的二分法获取中间数
int mid = ((right - left) >> 1) + left;
/**重点,若目标值小于mid,我们在折
*半的同时,保持让ans「插入位置」在右边界的右边
*因为若左边界=0,我们是无法在-1位置插入的
*/
if(target == nums[mid])
return mid;
if (target < nums[mid]) {
ans = mid;
right = mid - 1;
} else {
left = mid + 1;
}
}
return ans;
}