[总结]二分法(二分查找)
一、关于二分法
二分法是一种很平常却又很重要的算法。二分法能为我们解题时提供很大的帮助。
1. 使用前提
二分法的适用条件是序列具有二分性
,也就是单调性。当序列具有二分性,这时我们不断枚举区间中点才能判断这个值是否题设条件。
当题目中出现诸如最大值的最小
,最小值的最大
的问题时,答案具有二分性。
2. 分类
从二分的对象来分类,我们既可以二分最终的答案,我们也可以二分进行判断。
从二分的类型来分类,可以分为整数域上的二分,以及实数域上的二分。
3. 易错点
二分法简单易写,但是却很容易写错。我们有很多方法实现二分,而其中的细节地方需要仔细考虑。
对于整数域上的二分:
我们需要注意终止条件,左右区间位置的变化,避免错过答案或造成死循环。
对于实数域上的二分:
我们需要注意精度的控制。
建议自己形成固定的代码模型,避免造成不必要的错误。
4. 二分法的延伸
C++ STL中的lower_bound
,upper_bound
也可以解决实现在一个序列中二分查找某个整数k
的后继。
二分法能够解决单调问题,进一步地,我们可以扩展二分法至三分法
。此时三分法可以解决单峰函数的极值问题。
二、整数域上的二分
1. 模板
在这里给出一种常见的模板:
while(l<=r){ int mid=(l+r)>>1; if(check(mid)){ ans=mid; r=mid-1; } else l=mid+1; }
二、实数域上的二分
1. 模板
实数域上的二分相对简单,只要r-l
到达我们所需的精度即可。
#define eps 1e-5 while(r-l>eps){ double mid=(l+r)/2; if(check(mid)) r=mid; else l=mid; }
当我们不确定精度的时候,我们可以采用循环固定次数的形式进行计算。一般这种方式得到的结果的精度比设置的eps更高:
for(int i=1;i<=100;i++){ double mid=(l+r)/2; if(check(mid)) r=mid; else l=mid; }
三、练习
例1:#9100055「一本通 1.2 例 1」愤怒的牛 / SP297 AGGRCOW - Aggressive cows / P1316 丢瓶盖
分析:
很基础的二分,每次二分牛的间隔,如果能放下这c头牛,那么继续扩大这个距离,否则缩小这个距离,直到找到答案。
代码如下:
#include<bits/stdc++.h> using namespace std; int f[1000050],n,c,rem; int judge(int x){ int num=0; int temp=f[1]; for(int i=2;i<=n;i++){ if(f[i]-temp<x) num++; else temp=f[i]; if(num>rem) return 0; } return 1; } int main() { scanf("%d%d",&n,&c); for(int i=1;i<=n;i++) scanf("%d",&f[i]); sort(f+1,f+n+1); rem=n-c; int maxn=0; int l=1,r=f[n]-f[1]; while(l+1<r){ int mid=(l+r)/2; if(judge(mid)) l=mid; else r=mid; } printf("%d\n",l); return 0; }
例2:P1661 扩散
分析:
并查集+二分答案。二分枚举形成一个连通块的时间,每次使用并查集统计,如果最后集合的数量大于1,那么移动左区间,否则移动右区间。注意两个点都会扩张,因此单位时间会走双倍的距离。
代码如下:
#include<bits/stdc++.h> #define N 100010 using namespace std; int sx[N],sy[N],pre[N],n; int Find(int x){ return (pre[x]==x)? x:Find(pre[x]); } int check(int mid){ for(int i=1;i<=n;i++) pre[i]=i; for(int i=1;i<=n;i++){ for(int j=i+1;j<=n;j++){ int dis=abs(sx[i]-sx[j])+abs(sy[i]-sy[j]); if(mid*2>=dis){ int fi=Find(i); int fj=Find(j); if(fi!=fj) pre[fi]=fj; } } } int cnt=0; for(int i=1;i<=n;i++) if(pre[i]==i) cnt++; return (cnt==1)? 1:0; } int main() { scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d%d",&sx[i],&sy[i]); int l=0,r=1e9,ans=0; while(l<=r){ int mid=(l+r)>>1; if(check(mid)){ r=mid-1; ans=mid; } else l=mid+1; } printf("%d",ans); return 0; }
例3:P1182 数列分段 Section II
分析:
二分枚举每段总和为mid时是否可行,分了超过m段就更新左区间,否则更新右区间。
代码如下:
#include<bits/stdc++.h> using namespace std; int a[100010]; int n,m,ans,l,r; int judge(int mid){ int sum=0,cnt=1; for(int i=1;i<=n;i++){ if(sum+a[i]<=mid) sum+=a[i]; else{ sum=a[i]; cnt++; } } return (cnt<=m)?1:0; } int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++){ scanf("%d",&a[i]); l=max(l,a[i]); r+=a[i]; } while(l<=r){ int mid=(l+r)>>1; if(judge(mid)){ ans=mid; r=mid-1; } else l=mid+1; } printf("%d",ans); return 0; }
例4:POJ 2018 Best Cow Fences
分析:
实数域上的二分。因为平均值只是描述数与数的离散关系,所以我们同时加或减对整个序列的平均值都没有影响。所以我们对序列减去平均值后,问题化为存不存在这样的序列使得区间和大于0。
我们在\(O(N)\)复杂度内使用前缀和做减法处理出全序列中最大的一段子序列,如果此时序列和小于0,那么我们枚举的平均值过大,因此缩进右区间,反之同理。
#include<bits/stdc++.h> #define N 100010 #define INF 1e10 using namespace std; double a[N],b[N],sum[N]; int main() { int n,len; scanf("%d%d",&n,&len); for(int i=1;i<=n;i++) scanf("%lf",&a[i]); double l=-1e6,r=1e6; double dlt=1e-5; while(r-l>dlt){ double mid=(l+r)/2; for(int i=1;i<=n;i++) b[i]=a[i]-mid;//削去平均值 for(int i=1;i<=n;i++) sum[i]=sum[i-1]+b[i];//求前缀和 double ans=-INF,temp=INF; for(int i=len;i<=n;i++){ temp=min(temp,sum[i-len]);//因为长度大于等于L,所以确定一个min左端点 ans=max(ans,sum[i]-temp); } if(ans>=0) l=mid;//可以达到该平均值 else r=mid; } printf("%d",int(r*1000)); return 0; }
例5:CF670C Cinema
分析:
贪心思想,二分枚举每场电影能听懂配音的人数以及看懂字幕的人数,首先满足听懂配音人数,其次满足看懂字幕的人数。
#include<bits/stdc++.h> #define N 200010 using namespace std; inline void read(int &x){ x=0;int flag=1;char ch=getchar(); while(ch<'0'||ch>'9'){ if(ch=='-') flag=-1; ch=getchar(); } while(ch>='0'&&ch<='9'){ x=(x<<1)+(x<<3)+ch-'0'; ch=getchar(); } x*=flag; } int n,m,a[N],b[N],v,pow1,pow2,last1,last2; int ans=1; int main() { read(n); for(int i=1;i<=n;i++) read(a[i]); sort(a+1,a+n+1);read(m); for(int i=1;i<=m;i++) read(b[i]); for(int i=1;i<=m;i++){ read(v); int pow1=(upper_bound(a+1,a+n+1,b[i])-a-1)-(lower_bound(a+1,a+n+1,b[i])-a-1); int pow2=(upper_bound(a+1,a+n+1,v)-a-1)-(lower_bound(a+1,a+n+1,v)-a-1); if(pow1>last1||(pow1==last1&&pow2>last2)) last1=pow1,last2=pow2,ans=i; } printf("%d",ans); return 0; }
例6:POJ3579 Median
代码如下:
#include<iostream> #include<cstdlib> #include<cstdio> #include<algorithm> const int N=1e6; using namespace std; int a[N],n,m,ans; int check(int val) { int cnt=0; for(int i=1;i<=n;i++) cnt+=n-(lower_bound(a+1,a+n+1,a[i]+val)-a-1); if(cnt>m) return 1; else return 0; } int main() { while(~scanf("%d",&n)) { m=n*(n-1)/4; ans=-1; for(int i=1;i<=n;i++) scanf("%d",&a[i]); sort(a+1,a+n+1); int l=1,r=a[n]-a[1]; while(l<=r){ int mid=(l+r)>>1; if(check(mid)){ ans=mid; l=mid+1; } else r=mid-1; } printf("%d\n",ans); } return 0; }
例7:P1083 借教室
分析:
一道很好的思维题,运用了差分的思想。我们在第i天借了k个教室时,在这个时间节点累加这k个教室,在第j天归还的时候再减去。这样我们就能知道任意一天借出教室的数量。我们不断二分这个不能满足的日期,如果最后结果为m,那么全都能满足。否则二分终止的位置就是不能满足的日期。
代码如下:
#include<bits/stdc++.h> using namespace std; int num[1000010],day[1000010]; int m,n,l[1000010],r[1000010],req[1000010]; int judge(int mid){ memset(day,0,sizeof(day)); for(int i=1;i<=mid;i++){ day[l[i]]+=req[i]; day[r[i]+1]-=req[i]; } if(day[1]>num[1]) return 0; for(int i=2;i<=n;i++){ day[i]+=day[i-1]; if(day[i]>num[i]) return 0; } return 1; } int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) scanf("%d",&num[i]); for(int i=1;i<=m;i++) scanf("%d%d%d",&req[i],&l[i],&r[i]); int L=1,R=m,ans=0; while(L<=R){ int mid=(L+R)>>1; if(judge(mid)){ L=mid+1; } else R=mid-1,ans=mid; } if(R!=m) printf("-1\n%d",ans); else printf("0"); return 0; }