题目描述
给定一个排序的整数数组 nums 和一个整数目标值 target ,请在数组中找到 target ,并返回其下标。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
题目来源:LCR 搜索插入位置
知识准备
二分查找算法
public static int binarySearchBasic(int[] arr, int target) {
int i = 0; //指向数组头部
int j = arr.length - 1; //指向数组尾部
while (i <= j) {
//取i和j的中间
int mid = (i + j) >>> 1;
if (target < arr[mid]) { //目标在左边
j = mid - 1;
} else if (arr[mid] < target) { //目标在右边
i = mid + 1;
} else { //找到了!
return mid;
}
}
//找不到目标元素则返回-1
return -1;
}
心细的孩子肯定发现了为何取中不是**(i + j) / 2**,而是**(i + j) >>> 1**。
这里涉及到了int基本数据类型的取值范围,假设
i = 1;
j = Integer.MAX_VALUE;
i、j相加所得值超过了int取值范围,计算机采取补码形式对该值进行存储。所以这次选择 (i + j) >>> 1 。
除此之外,作者已知的形式还有分式因解int mid = i/2 + j/2,还有 int mid = ((right - left) >> 1) + left。
基础版的二分查找的对象是一个有序数组,找到则返回mid,没找到则返回-1。
作者觉得返回 -1 太单调了,不如返回 i ?通过作者发现:
在找到目标元素的情况下,i 等于与目标元素相等(在mid右边)的最靠左索引;
在没找到目标元素的情况下,i就只好等于比目标元素大(在mid右边)的最靠左索引。
二分查找修改版
由此可知,要想知道目标元素的插入位置,只需要返回 i 即可
public static int binarySearchBasic(int[] arr, int target) {
int i = 0; //指向数组头部
int j = arr.length - 1; //指向数组尾部
while (i <= j) {
//取i和j的中间
int mid = (i + j) >>> 1;
if (target < arr[mid]) { //目标在左边
j = mid - 1;
} else if (arr[mid] < target) { //目标在右边
i = mid + 1;
} else { //找到了!
return mid;
}
}
//返回i的缘故?又为何会比返回-1更有用?
//1.找到的情况,i表示与目标相等的最靠左的索引
//2.没找到的情况,i表示比目标大的最靠左的索引
return i;
}
做题心得
1.菜鸟作者的一次尝试,如若有不懂的可以评论区发问,如果有大神提出高见更好了!
2.相信读者们都发现了注解中的一句话【找到的情况,i表示与目标相等的最靠左的索引】,那是不是可以理解该代码也可以针对一个有重复元素的升序数组,在其中寻找重复元素的第一次出现的索引呢?