[数据结构] 二分查找 (四种写法)

最新推荐文章于 2024-08-17 20:04:47 发布

cn--萍诚1号

最新推荐文章于 2024-08-17 20:04:47 发布

阅读量1.2k

点赞数 2

文章标签：数据结构算法 c++ Powered by 金山文档

原文链接：https://www.cnblogs.com/MAKISE004/p/17093253.html

版权

文章详细介绍了二分查找算法的工作原理，包括在有序数组中查找目标值、找到目标值的第一个出现位置以及最后一个出现位置的方法。通过递归和迭代两种方式展示了二分查找的代码实现，并提到了在查找终止位置时处理特殊情况的技巧。

摘要由CSDN通过智能技术生成

二分查找（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二分查找图解

（1）

（2）

（3）

1|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;}

2|0二分查找递归

上面二分查找也可以写成递归的形式。大致步骤为：

（1）当前层 mid 位置元素等于目标值 target，直接 return mid；

（2）如果小于目标值，递归搜索 [mid + 1, right] 范围；

（3）如果大于目标值，递归搜索 [left, mid - 1] 范围。

//二分查找法递归写法int BinarySearchRec(vector<int> &a, intleft, intright, inttarget) { 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;}

3|0二分查找元素起始位置

3|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 。

3|2查找等于目标值起始位置图解

（1）

（2）

（3）

（4）

3|3查找等于目标值起始位置代码

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;}

4|0二分查找元素终止位置

4|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。

但是要注意的是，在这里取 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 操作后向上取整就可以解决这个问题。

4|2查找等于目标值终止位置图解（取mid向上取整）

（1）

（2）

（3）

（4）

4|3查找等于目标值终止位置图解（取mid向下取整）*

（1）

（2）

（3）

（4）

4|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;}

5|0相关试题及完整程序

5|1相关试题

Acwing 789.数的范围

Leetcode 34.在排序数组中查找元素的第一个和最后一个位置

5|2完整程序

#include<iostream>#include<vector>using namespace std;
//二分查找int BinarySearch(vector<int> &v, inttarget){    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, intleft, intright, inttarget) {    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, inttarget){    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, inttarget){    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;}