以查找某个特定元素在已排序数组中的索引为例,且此数组为从小到大排序。
算法思路:
1、取第一个索引和最后一个索引代表两个游标,一个lo,一个hi
2、取lo和hi的中间值,即为mid,如下图所示,中间值为4
3、取出中间索引处的值,与输入值i比较
- 假如比i大,那么说明我们要的值在上面,把hi值变成刚刚取的中间索引值,即hi=mid-1,lo值保持不变,
- 假如比i小,那么说明我们要的值在下面,把lo值变成刚刚取得中间索引值,即lo=mid+1,hi值保持不变
- 假如两个值相等,那么返回mid
注1:为什么需要mid减1或加1?
4、重复2~3步,直到lo>hi
注2:为什么是>而不是>=?
普通实现:
public static int indexOf(int i, int[] a) {
int lo = 0;//定义上索引
int hi = a.length - 1;//定义下索引
while(lo<=hi) {
int mid = (lo + hi) / 2;
if(a[mid]>i) hi = mid - 1;//假如比i大
else if(a[mid]<i) lo = mid + 1;//假如比i小
else return mid;
}
return -1;//没找到
}
递归实现:
public static int indexOf(int i, int[] a) {
indexOf(i, a, 0, a.length-1);
}
public static int indexOf(int i, int[] a, int lo, int hi) {
if(lo>hi) return -1;
int mid = (lo + hi) / 2;
if(a[mid]>i) return indexOf(i, a, lo, mid-1);
else if(a[mid]<i) return indexOf(i, a, mid+1, hi);
return mid;
}
注1解答:因为mid已经不满足条件,我们已经不需要它了
注2解答:因为在=出现的时候,mid = lo = hi,而这个mid值是上一步的mid减1或加1得到的,即这两个mid不可能相等,为了验证最后的这个mid是不是我们需要的mid,所以还需要这两个值相等,而下一步要么是lo = mid + 1 > hi,要么是hi = mid - 1 < lo,都会不满足lo<=hi的条件而跳出循环