绪论
最近一直在刷二分查找的题目,发现二分查找有很多种题目,使用常规的二分查找做这些题目不太方便,所以我刷题的时候总结了一种更方便更易于理解二分法来解决这些问题。
主要思路是通过修改while循环中begin和end的赋值条件来最终使begin和end同时指向目标位置,这个目标位置可以是等于某一个目标值,也可以是满足/不满足某个条件的第一个元素。相信我,看完这一篇可以让你对二分查找有个透彻的了解!看完记得点个赞哦❤。
左闭右开区间使用二分法查找满足条件的第一个位置
我一般使用C++刷题,力扣上面给的数组容器都是vector,vector的end()返回的是尾后元素,所以我习惯使用左闭右开区间的解法,这种解法是我在看C++ Primer时学会的,基本能够适用于所有情况,并且左闭右闭区间也能够转换成左闭右开区间再查找。
假如给定一个nums和一个目标值val,查找容器中第一个大于等于val的值,可以使用如下方法:
#include<iostream>
#include<vector>
using namespace std;
//返回第一个大于等于val的元素的迭代器,如果所有元素都小于val,那么会返回nums.end()
vector<int>::iterator binarySearch(vector<int>& nums, int val)
{
auto begin = nums.begin();
auto end = nums.end();
vector<int>::iterator mid;
while (begin != end)
{
mid = begin + (end - begin) / 2;
if (*mid >= val)end = mid;
else
begin = mid + 1;
}
//到这里begin = end = 目标迭代器
return begin;
}
while循环的结束条件是(begin != end),这是因为区间[begin,end)在begin=end时是没有元素的,所以到这里循环结束,begin与end收敛到一点,该点就是满足条件的第一个点。有的小伙伴可能会问:“使用begin!=end作为结束条件,会不会当begin>end时陷入死循环”。答案是不会的,后面分析迭代过程时会讲到原因。
另外,要理解为什么这个点就是满足条件的第一个点,简单来说,由于end只有在mid满足条件时才会移动到mid,所以end指向的元素一定是符合条件的;当mid不满足条件时,begin指向mid的下一个位置,所以begin是从不满足条件的区间向满足条件的区间移动的(begin = mid + 1是begin能够从不满足条件的区间移动到满足条件区间的原因)。于是begin从不满足条件的区间向目标位置移动并且能够跨越到满足条件的区间,end从满足条件的位置向目标位置移动,但是他只能指向满足条件的位置,二者最终相等,且指向的一定是满足条件的第一个位置。这里的条件是*mid>=val,你也可以自己写一个回调函数作为判断条件来增加代码灵活性,但是前提是nums对于你的条件必须是有序的。
迭代过程分析
总体迭代过程
如果你已经熟练使用这种方法了,就使用我前面简单来说的内容记忆并使用就好。但为了更方便大家理解,我还是讲一下上面函数的迭代过程。当元素还很多时,每一次迭代的过程大概如下所示。
可以看出,如果满足条件,end在向满足条件的第一个位置靠拢;如果不满足条件,begin也在向满足条件的第一个位置靠拢。并且由于begin时可以跳出不满足条件的区间的,所以最后begin与end都会指向第一个满足条件的区间。
由于mid在这次循环中是被检查过的,所以下一次循环左边或右边的子区间都不会包括这次循环中mid指向的元素。
最终迭代过程
这个迭代过程比较难理解的地方在于最后还剩两个或者三个元素的时候,所以我前面一直在粗略地讲为什么他们最终共同指向满足条件的第一个位置。如果搞清楚了迭代的过程,你就可以修改代码来寻找其他位置;也会明白为什么这种方法不会让begin>end而有的写法又会。由于该函数能够返回尾后元素,并且一种极端的情况是当最终begin=end=nums.end()时,循环不会执行、也就不会对尾后元素解引用,所以这里不把尾后元素与普通元素做区分。
begin=end前的最后一步会有三种情况,其中两个元素有两种:a. *mid >= val;b. *mid < val;三个元素只有一种情况:c.*mid < val,因为三个元素且当*mid >= val时会成为情况a或者b。
情况a
当只有两个元素时,由于mid = begin + (end - begin) / 2;所以mid会指向begin在的位置,而end最终来到begin在的位置上,循环结束。
情况b
当只有两个元素且*mid < val时,begin会跳到mid的下一个位置,即end所在的位置,循环结束。
情况c
这种情况下begin会移动两个位置,到达end的位置 ,循环结束。
可以看到所有的情况下都没有begin > end的情况,所以使用begin != end作为循环的终止条件是合理的。那么什么情况下begin会有可能大于end呢?这种情况在情况a且满足条件时end迭代为:end=mid-1时会发生。这种写法常见于左闭右闭区间中,因为左闭右闭区间产生的左边区间或右边区间也要是左闭右闭的,而mid已经检查过了,所以两边新区间都不会包括mid,就会有这种写法。因此左闭右闭区间的终止条件常常是(begin <= end),begin = end时区间内是有一个元素的,所以也需要进行一次循环来检查。
上面方法是将查找某个元素和查找满足条件的第一个位置统一起来的写法,如果你只是需要查找某一个元素是否存在,也可以在*mid等于该元素时直接return来减少循环次数。
查找满足条件的最后一个位置
了解以上原理后就能知道查找满足条件的最后一个位置的方法。只要反过来操作就行了,反过来可以利用下标,也可以利用反向迭代器。
利用下标
69.x的平方根:查找符合条件的最后一个位置;
class Solution {
public:
bool condition(int n,int x)
{
return n <= x/n;
}
int mySqrt(int x) {
int begin = 0,end = x,mid;
while(begin != end)
{
mid = end - (end - begin) / 2;
if(condition(mid,x)) begin = mid;
else end = mid-1;
}
return begin;
}
};
利用反向迭代器
反向迭代器的rbegin指向的容器中最后一个元素,rend指向的容器中第一个元素的前一个元素,+操作会从后向前移动。反向迭代器使我们能使用与正向迭代器极其相似的方法进行操作。
34.在排序数组中查找元素的第一个和最后一个位置。
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
//使用正向迭代器求开始元素
vector<int>::iterator begin = nums.begin(), end = nums.end(), mid;
while (begin != end)
{
mid = begin + (end - begin) / 2;
if (*mid >= target)end = mid;
else begin = mid + 1;
}
if (begin == nums.end() || *begin != target)return { -1,-1 };
//使用反向迭代器求介绍元素
vector<int>::reverse_iterator rbegin = nums.rbegin(), rend = nums.rend(), rmid;
while (rbegin != rend)
{
rmid = rbegin + (rend - rbegin) / 2;
if (*rmid <= target)rend = rmid;
else rbegin = rmid + 1;
}
int i = begin - nums.begin();
int j = nums.rend() - rbegin - 1;
return{ i,j };
}
};
当然这个题也可以用下标的方法,这样就不需要迭代器到下标的转换,不过会比反向迭代器更复杂一点。
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
//求第一个元素
int begin = 0, end = nums.size(), mid;
while (begin != end)
{
mid = begin + (end - begin) / 2;
if (nums[mid] >= target)end = mid;
else
begin = mid + 1;
}
if (begin == nums.size() || nums[begin] != target)return { -1,-1 };
//求最后一个元素
int rbegin = -1, rend = nums.size() - 1, rmid;
while (rbegin != rend)
{
rmid = rend - (rend - rbegin) / 2;
if (nums[rmid] <= target)rbegin = rmid;
else
rend = rmid - 1;
}
return{ begin,rbegin };
}
};
leetcode相关题目
如果上面的看懂了,可以尝试力扣:
704.二分查找:最基础的二分查找;
35.搜索插入位置:查找满足条件的第一个位置;
69.x的平方根:查找符合条件的最后一个位置;
34.在排序数组中查找元素的第一个和最后一个位置。