二分查找也称折半查找(Binary Search),它是一种效率较高的查找方法。但是,折半查找要求线性表必须采用顺序存储结构,而且表中元素按关键字有序排列。
二分查找的概念比较易懂,但是在实现上总是会边界考虑不周或陷入死循环的问题存在。下面介绍一种易于理解且放之四海而皆准的实现方法(不管要找的是<,<=,>,>=,都可以套用模板方法)。
先放伪代码
public int searchInsert(int[] nums, int target) {
int l = -1;
int r = nums.length;
while(l+1 != r) {
int mid = (l + r) / 2;
if(isBlue(mid)) {
l = mid;
} else {
r = mid;
}
}
return l or r;
}
例如上面的有序数组,在二分查找的问题中会有如下几种情况:
- 第一个等于3的索引
- 最后一个等于3的索引
- 找不到对应的值,则返回需要插入的索引位置
其实这几个问题都可以归纳为查找某个边界,获取该边界左侧或者右侧的索引值。
例如:
- 查找第一个等于3的索引,则我们先找到数值2、3的边界,即索引1和2之间,那么右侧的索引值就是我们想要的;
- 查找最后一个等于3的索引,则我们先找到数值3、4的边界,即索引4、5之间,那么左侧的索引值就是我们想要的;
- 如果我们要找2.5,发现没找到,但是我们可以找到索引1、2之间,那么右侧的索引值就是要插入的位置。
上面给出的模板代码中,isBlue方法会根据查询条件的不同而不同
- 查找第一个等于3的索引:mid < 3, 最后return r
- 查找最后一个等于3的索引:mid <= 3,最后return l
这样的实现方法,会导致死循环吗?答案是不会。如下通过数学递归法来证明
- 如果l+1=r,那么会退出while循环,不会出现死循环
- 如果l+2=r,那么下次循环会变更成第一种情况,不会出现死循环
- 如果l+3=r,那么下次循环会变更成第一种或第二种情况,不会出现死循环
- 以此类推,其他情况最终都会更变为第一种或第二种情况,不会出现死循环
参考: