二分查找
二分查找
二分查找(Binary Search)也叫作折半查找,前提是查找的顺序结构是有序的,我们一般在数组上进行二分查找。
二分查找就好像猜数字大小游戏一样。假设要数字目标值属于 [1, 1000] 范围内,当我们猜的数字小于这个目标值时("Too low"),我们需要往大去猜;反之大于这个目标值时("Too high"),我们需要往小去猜。当然这里猜的方式并不是盲目的,我们每次都取中间值去猜,每猜一次可以缩小当前一半的范围,这样可以大大提高效率。二分查找本质上也是这样的过程,时间复杂度为 O(logn) ,在较大数据情况下相比线性查找要快非常多。
我们定义一个左指针 left 标记当前查找区间的左边界,一个右指针 right 标记当前查找范围的右边界。每次取 mid 来判断当前取的值是否等于目标值 target。如果等于 target 就直接返回 mid ;如果小于目标值 target ,那么将左边界变为 mid + 1,缩小区间范围继续在 [mid + 1, right] 范围内进行二分查找,如果大于目标值 target ,那么将右边界变为 mid - 1,缩小区间范围继续在 [left, mid - 1] 范围内进行二分查找。
假如最后出现了 left > right 的情况,说明区间范围大小缩小到 0 都无法找到该目标值,那么很明显数组中不存在这个目标值 target,此时退出循环,返回 -1 。
二分查找图解
(1)
(2)
(3)
二分查找代码
//二分查找
int BinarySearch(vector<int> &v, int target){
int left = 0, right = v.size() - 1;
while(left <= right){
int mid = (left + right) >> 1;
if(v[mid] == target)
return mid;
if(v[mid] < target)
left = mid + 1;
else
right = mid - 1;
}
return -1;
}
二分查找 递归
上面二分查找也可以写成递归的形式。大致步骤为:
(1)当前层 mid 位置元素等于目标值 target,直接 return mid;
(2)如果小于目标值,递归搜索 [mid + 1, right] 范围;
(3)如果大于目标值,递归搜索 [left, mid - 1] 范围。
//二分查找法递归写法
int BinarySearchRec(vector<int> &a, int left, int right, int target) {
int mid = (left + right) >> 1;
if(left <= right){
if(a[mid] == target)
return mid;
if(a[mid] < target)
return BinarySearchRec(a, mid + 1, right, target);
else
return BinarySearchRec(a, left, mid - 1, target);
}
return -1;
}
二分查找 元素起始位置
查找等于目标值的第一个位置
数组中可能存在连续的多个位置元素都等于目标值 target ,当我们要查找第一个出现的位置,我们需要保证查找到位置的左边所有元素都满足小于 target,右边所有元素都满足大于等于 target。
当出现 a[mid] < target 时,说明我们要查找的位置一定在 [mid + 1, right] 范围内,当然也可以写成 (mid, right] ;当出现 a[mid] >= target 时,说明要查找的位置有可能是当前的 mid,也有可能是当前 mid 左边的某个位置,所以此时要查找的位置一定在 [left, mid] 范围内。
因此,当 a[mid] < target,将 left = mid + 1 ;当 a[mid] >= target,将 right = mid。
当最后 left == right 即两个指针相遇时退出循环,最后要判断一下相遇位置处的元素是否等于目标值 target。如果等于目标值,就返回 left 或者 right,如果不等于目标值,说明不存在该元素,那么就返回 -1 。
查找等于目标值起始位置图解
(1)
(2)
(3)
(4)
查找等于目标值起始位置代码
int BinarySearchfirst(vector<int> &v, int target){
int left = 0, right = v.size() - 1;
while(left < right){
int mid = (left + right) >> 1;
if(v[mid] < target)
left = mid + 1;
else
right = mid;
}
return v[left] == target ? left : -1;
}
二分查找 元素终止位置
查找等于目标值的最后一个位置
当我们要查找最后一个出现的位置,我们需要保证查找到位置的左边所有元素都满足小于等于 target,右边所有元素都满足大于 target。
当出现 a[mid] > target 时,说明我们要查找的位置一定在 [left, mid - 1] 范围内,当然也可以写成 [left, mid) ;当出现 a[mid] <= target 时,说明要查找的位置有可能是当前的 mid,也有可能是当前 mid 右边的某个位置,所以此时要查找的位置一定在 [mid, right] 范围内。
因此,当 a[mid] > target,将 right = mid - 1 ;当 a[mid] <= target,将 left = mid。
当最后 left == right 即两个指针相遇时退出循环,最后要判断一下相遇位置处的元素是否等于目标值 target。如果等于目标值,就返回 left 或者 right,如果不等于目标值,说明不存在该元素,那么就返回 -1 。
但是要注意的是,在这里取 mid 时,我们不能和之前一样取 (left + right) >> 1,而要采取 (left + right + 1) >> 1 的形式。我们可以假设只有数组两个元素,两个元素都等于目标值,显然此时我们要找的最后一个位置为下标1。我们模拟一下,初始情况下 left 为 0,right 为 1,如果采用 (left + right) >> 1,那么此时的 mid 就等于 0,这个时候出现了 left 依旧等于之前 left 的情况,那么显然这个时候区间无法进行缩小,left 会一直等于 0,这个时候就陷入死循环了。
我们仔细看一下,当前这种情况的特点是 left + 1 == right,那么我们取 mid 时:
mid = (left + right) >> 1 = (2 * left + 1) >> 1 = left,很明显left 会一直等于 mid。
如果我们能够让 left 在这种 left + 1 == right 情况下使得 left 取到 right 即往后一位,那么我们的区间范围就得以缩小,也不会陷入死循环。所以我们采用 (left + right + 1) >> 1 取 mid,问题就得以解决了。此时:
mid = (left + right + 1) >> 1 = (2 * left + 2) >> 1 = left + 1 = right,可以看出 left 往后了一位。
归根究底还是因为整形数据除 2 会自动进行向下取整的问题,进行 +1 操作后向上取整就可以解决这个问题。
查找等于目标值终止位置图解(取mid向上取整)
(1)
(2)
(3)
(4)
查找等于目标值终止位置图解(取mid向下取整)*
(1)
(2)
(3)
(4)
查找等于目标值终止位置代码
int BinarySearchlast(vector<int> &v, int target){
int left = 0, right = v.size() - 1;
while(left < right){
int mid = (left + right + 1) >> 1;
if(v[mid] > target)
right = mid - 1;
else
left = mid;
}
return v[left] == target ? left : -1;
}
相关试题及完整程序
相关试题
Acwing 789.数的范围
Leetcode 34.在排序数组中查找元素的第一个和最后一个位置
完整程序
#include<iostream>
#include<vector>
using namespace std;
//二分查找
int BinarySearch(vector<int> &v, int target){
int left = 0, right = v.size() - 1;
while(left <= right){
int mid = (left + right) >> 1;
if(v[mid] == target)
return mid;
if(v[mid] < target)
left = mid + 1;
else
right = mid - 1;
}
return -1;
}
//二分查找法递归写法
int BinarySearchRec(vector<int> &a, int left, int right, int target) {
int mid = (left + right) >> 1;
if(left <= right){
if(a[mid] == target)
return mid;
if(a[mid] < target)
return BinarySearchRec(a, mid + 1, right, target);
else
return BinarySearchRec(a, left, mid - 1, target);
}
return -1;
}
//查找元素起始位置
int BinarySearchfirst(vector<int> &v, int target){
int left = 0, right = v.size() - 1;
while(left < right){
int mid = (left + right) >> 1;
if(v[mid] < target)
left = mid + 1;
else
right = mid;
}
return v[left] == target ? left : -1;
}
//查找元素终止位置
int BinarySearchlast(vector<int> &v, int target){
int left = 0, right = v.size() - 1;
while(left < right){
int mid = (left + right + 1) >> 1;
if(v[mid] > target)
right = mid - 1;
else
left = mid;
}
return v[left] == target ? left : -1;
}
int main(){
vector<int> v = {7,7,7,7};
cout<<BinarySearchFirst(v, 7)<<endl;
cout<<BinarySearchLast(v, 7)<<endl;
cout<<BinarySearchRec(v, 0, v.size() - 1, 7)<<endl;
cout<<BinarySearch(v, 7)<<endl;
}
评论区