今天刷LeetCode的时候,遇到了一道二分查找的题目。代码轮廓很容易就写出来了,但是,调边界条件却调了很久。原因还是因为自己对二分查找掌握的不够透彻。于是又查了一些资料,看了别人的一些博客,再次学习了二分查找~下面总结今天看的七种二分查找的类型,代码轮廓大致一样,大多是是判断条件不一样。
注: 以下讨论全是在不严格递增数组下实现的,例如,array[] = {1,2,2,3,3,4,4,5,6,7}
1. 基础版——查找有序数组中key的下标
查找有序数组中key的下标,若存在多个,则返回任意一个下标,不存在该数的话则返回-1。
解析: 当我们找到array[mid] == key
时,我们应该返回mid的值。当我们找到array[mid] > key
时,那么我们需要求的元素应该在array[mid]
之前,所以将end
指针赋值为mid - 1
。当我们找到array[mid] < key
时,那么我们需要求的元素应该在array[mid]
之后,所以将start
指针赋值为mid + 1
。
//非递归版
int binSearchIndex(int array[] , int len , int key){
int start = 0 , end = len - 1 , mid = 0;
while(start <= end){
mid = start + ((end - start ) >> 1);
if(array[mid] == key){
return mid;
}else if(array[mid] < key){
start = mid + 1;
}else{
end = mid - 1;
}
}
return -1;
}
//递归版
int binSearchIndexRecursion(int array[] , int start , int end , int key){
int mid = (start + end) >> 1;
if(array[mid] == key){
return mid;
}
if(array[mid] > key){
binSearchRecursion(array , start , mid - 1 , key);
}else if(array[mid] < key){
binSearchRecursion(array , mid + 1 , end , key);
}
}
2. 基础版——查找有序数组中第一个大于等于key的下标
查找有序数组中第一个大于等于key的下标,如果不存在,则返回-1。
解析: 当我们找到array[mid] >= key
时,我们应该假设它不是第一个大于等于key
的元素,那么我们需要求的元素应该在array[mid]
之前,所以将end
指针赋值为mid - 1
。当我们找到array[mid] < key
时,那么我们需要求的元素应该在array[mid]
之后,所以将start
指针赋值为mid + 1
。这里我们应该判断一下最后返回的start
值有没有超过数组的长度,如果超过了,说明不存在大于等于key
的元素,应该返回-1。
//非递归版本
int binSearch(int array[] , int len , int key){
int start = 0 , end = len - 1 , mid = 0;
w