二分查找模板

一、查找精确值

从一个有序数组中找到一个符合要求的精确值(如猜数游戏)。如查找值为Key的元素下标,不存在返回-1。

 1 //这里是left<=right。
 2 //考虑这种情况:如果最后剩下A[i]和A[i+1](这也是最容易导致导致死循环的情况)首先mid = i,
 3 //如果A[mid] < key,那么left = mid+1 = i +1,如果是小于号,则A[i + 1]不会被检查,导致错误
 4 int left = 1,right = n;
 5 while(left <= right)
 6 {
 7    //这里left和right代表的是数组下标,所有没有必要改写成mid = left + (right - left)/2;
 8   //因为当代表数组下标的时候,在数值越界之前,内存可能就已经越界了
 9   //如果left和right代表的是一个整数,就有必要使用后面一种写法防止整数越界
10         int mid = (left + right) / 2;
11     if(A[mid] == key)
12       return mid;
13     else if(A[mid] > key)//这里因为mid不可能是答案了,所以搜索范围都需要将mid排除
14       right = mid - 1;
15     else
16       left = mid + 1;
17 }
18 return -1;

二、查找大于等于/大于key的第一个元素
这种通常题目描述为满足某种情况的最小的元素。

 1 int left = 1,right = n;
 2 while(left < right)
 3 {
 4   //这里不需要加1。我们考虑如下的情况,最后只剩下A[i],A[i + 1]。
 5   //首先mid = i,如果A[mid] > key,那么right = left = i,跳出循环,如果A[mid] < key,left = right = i + 1跳出循环,所有不会死循环。
 6   int mid = (left + right) / 2;
 7   if(A[mid] > key)//如果要求大于等于可以加上等于,也可以是check(A[mid])
 8     right = mid;
 9   //因为找的是大于key的第一个元素,那么比A[mid]大的元素肯定不是第一个大于key的元素,因为A[mid]已经大于key了,所以把mid+1到后面的排除
10   else
11     left = mid + 1;
12   //如果A[mid]小于key的话,那么A[mid]以及比A[mid]小的数都需要排除,因为他们都小于key。不可能是第一个大于等于key的元素,
13 }

三、查找小于等于/小于key的最后一个元素

这种通常题目描述为满足某种情况的最大的元素。如Leetcode69题,求sqrt(x)向下取整就是这种模板。

 1 int left = 1, right = n;
 2 while(left < right)
 3 {
 4   //这里mid = (left + right + 1) / 2;
 5   //考虑如下一种情况,最后只剩下A[i],A[i + 1],如果不加1,那么mid = i,如果A[mid] < key,执行更新操作后,left = mid,right = mid + 1,就会是死循环。
 6   //加上1后,mid = i + 1,如果A[mid] < key,那么left = right = mid + 1,跳出循环。如果A[mid] > key,left = mid = i,跳出循环。
 7   int mid = (left + right + 1) / 2;
 8   if(A[mid] < key)
 9     left = mid;//如果A[mid]小于key,说明比A[mid]更小的数肯定不是小于key的最大的元素了,所以要排除mid之前的所有元素
10   else
11     right = mid - 1;//如果A[mid]大于key,那么说明A[mid]以及比A[mid]还要大的数都不可能小于key,所以排除A[mid]及其之后的元素。
12 }

四、总结
最后两种情况的循环跳出条件是left<right,为什么不是小于等于呢?因为我们的区间变换思路是不断的舍去不可能是解的区间,最后只剩下一个数就是我们的解。而第一种情况就算最后只剩一个数也有可能不是解,所以需要使用小于等于。

查找精确值,循环条件是小于等于;查找满足情况的最大最小值,循环条件是小于。
查找满足条件的最大数,mid = (right + left + 1) / 2;查找满足条件的最小数,mid = (right + left)/2
mid = left + (right - left) / 2,不是适用于所有的情况。
如果存在没有解的情况,比如从[1,2,3,4,5]找出大于等于6的第一个数,我们只需要将最后剩下的数单独进行一次判断就可以了。

 

该文章出自于 https://www.acwing.com/blog/content/307/

posted @ 2019-07-10 20:52  巍萌先生  阅读(1956)  评论(0编辑  收藏  举报