还是把luogu上那篇搬过来吧qwq
何为整体二分?二分她儿子
先来看道题吧:
静态区间第(K)小:
给一个长度为(n)的序列(a),(m)次询问,每次询问用一个三元组表示((ql,qr,k)),即(a_{ql} ... a_{qr})中第(k)小的数是多少。(不一定要在线)
某神犇:主席树板子题,三分钟切了!
(Orz)。。。
因为我太菜了,所以不会主席树(qwq)。。。
那我们怎么办呢?
整体二分就横空出世了。。。
怎么个二分法呢?
我们先把所有询问放到一起。
然后二分一个(mid)(数值),同时令当前二分到的区间为([l,r])。。。
假设我们有左右两个用于放询问的空篮子(大雾
对于一组询问((ql,qr,k)),若(a_{ql} ... a_{qr})中比(mid)小的数大于或等于(k),那么这个询问的答案一定在([l,mid])里,我们就把他扔进左边的篮子里。
反之我们就把他扔进右边的篮子里。
这样就把询问分成了左右两种,同时二分的区间也变成了([l,mid])和([mid+1,r]),我们就可以递归做下去了。。。结束条件是(l=r),则这些询问的答案都是(l)。
是不是ylmb?
那么序列(a)里的数要怎么处理呢?
我们也把他看做一个二元组((i,x))表示(a_i = x)和询问放在一起。。。
那么在二分到(mid)的时候若有一个(a_i>mid),那么它和答案区间([l,mid])是半毛钱关系都没有的,我们就把他扔进([mid+1,r])里。
反之我们把他扔进左边的篮子。
那么怎么维护区间内比(mid)小的树呢?树状数组辣。。。做个前缀和就好了。当遇到了((i,x))时,如果(x<=mid),就在位置(i)上加(1)。
记住结束后要及时清空树状数组(尽量不要用(memset),如果数组很大的话(memset)还不如(for)循环)
时间复杂度:(O(nlog^2n)) (共递归(logn)层,一层的复杂度是(O(nlogn))的)
估计又是ylmb
还是看例题吧。。。
P3834 【模板】可持久化线段树 1(主席树)
就是上面的例子吧。。。
主席树模板题怎么能用主席树做呢?
具体实现(还是一个世纪前的码风):
#include <bits/stdc++.h>
#define IO(file) freopen(file".in","r",stdin),freopen(file".out","w",stdout)
#define debug(...) fprintf(stderr,__VA_ARGS__)
#define getchar nc
#define N 500010
#define INF 1e9
using namespace std;
inline char nc()
{
static char buf[1<<10],*p1=buf,*p2=buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<10,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
register int x=0,f=1; register char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f*=-1;ch=getchar();}
while(ch>='0'&&ch<='9')x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
return x*f;
}
template<class T> inline void write(T x,char end='
')
{
if(x==0) putchar('0'); if(x<0) x=-x,putchar('-');
static char buf[256]; register int top=0;
while(x) buf[++top]=x%10+48,x/=10;
while(top) putchar(buf[top--]);
putchar(end);
}
int n,m,cnt;
struct Query
{
int x,y,k;
int pos,type;
Query(){}
Query(int i,int j,int kk,int p,int t):x(i),y(j),k(kk),pos(p),type(t){}
}q[N],q1[N],q2[N];
int ans[N],c[N];
inline int lowbit(int x){return x&-x;}
inline void add(int x,int v)
{
for(;x<=n;x+=lowbit(x))
c[x]+=v;
}
inline int sum(int x)
{
int res=0;
for(;x;x-=lowbit(x))
res+=c[x];
return res;
}
void solve(int l,int r,int ql,int qr)
{
if(l>r || ql>qr) return;
if(l==r)
{
for(int i=ql;i<=qr;i++)
if(q[i].type) ans[q[i].pos]=l;
return;
}
int cnt1=0,cnt2=0,mid=(l+r)>>1;
for(int i=ql;i<=qr;i++)
{
if(q[i].type==0)
{
if(q[i].x<=mid) add(q[i].pos,q[i].k),q1[++cnt1]=q[i];
else q2[++cnt2]=q[i];
}
else
{
int tmp=sum(q[i].y)-sum(q[i].x-1);
if(q[i].k<=tmp) q1[++cnt1]=q[i];
else q[i].k-=tmp,q2[++cnt2]=q[i];
}
}
for(int i=1;i<=cnt1;i++)
if(q1[i].type==0) add(q1[i].pos,-q1[i].k);
for(int i=1;i<=cnt1;i++)
q[ql+i-1]=q1[i];
for(int i=1;i<=cnt2;i++)
q[ql+cnt1+i-1]=q2[i];
solve(l,mid,ql,ql+cnt1-1);
solve(mid+1,r,ql+cnt1,qr);
}
int main()
{
#ifdef LOCAL
IO("data");
#endif
n=read(),m=read();
for(int i=1;i<=n;i++)
{
int x=read();
q[++cnt]=Query(x,0,1,i,0);
}
for(int i=1;i<=m;i++)
{
int l=read(),r=read(),k=read();
q[++cnt]=Query(l,r,k,i,1);
}
solve(-INF,INF,1,cnt);
for(int i=1;i<=m;i++)
write(ans[i]);
return 0;
}
时间复杂度(O(nlog^2n)),好像比主席树(O(nlogn))慢一点。。。
实测下来:
主席树((804ms))
整体二分((1.28s))
好像是慢一点。。。
P2617 Dynamic Rankings
动态区间第(k)小。
看起来逼格很高的样子。。。
也有两种做法。。。
一种树状数组套主席树(这里不讲)
还有整体二分
两者的复杂度都是(O(nlog^2n))。
讲一下整体二分的做法。
把原数组里每个数看做((i,a_i,1)),表示如果(a_i<=mid)的话则位置(i)加一。
那么就可以很显然的把点修改看做两个这样的三元组((i,a_i,-1)),((i,t,1)),这样就可以解决修改操作了。
区间询问和上面类似,按顺序塞进(q)数组里,把原序列放在前面就好了。。。
#include <bits/stdc++.h>
#define getchar nc
#define debug(...) fprintf(stderr,__VA_ARGS__)
#define IO(file) freopen(file".in","r",stdin),freopen(file".out","w",stdout)
#define N 500010
#define INF 1e9
using namespace std;
inline char nc()
{
static char buf[1<<10],*p1=buf,*p2=buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<10,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
register int x=0,f=1; register char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f*=-1;ch=getchar();}
while(ch>='0'&&ch<='9')x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
return x*f;
}
template<class T> inline void write(T x,char end='
')
{
if(x==0) putchar('0'); if(x<0) x=-x,putchar('-');
static char buf[256]; register int top=0;
while(x) buf[++top]=x%10+48,x/=10;
while(top) putchar(buf[top--]);
putchar(end);
}
int n,m,cnt;
struct Query
{
int x,y,k;
int pos,type;
Query(){}
Query(int i,int j,int kk,int p,int t):x(i),y(j),k(kk),pos(p),type(t){}
}q[N],q1[N],q2[N];
int ans[N],c[N];
inline int lowbit(int x){return x&-x;}
inline void add(int x,int v)
{
for(;x<=n;x+=lowbit(x)) c[x]+=v;
}
inline int sum(int x)
{
int res=0;
for(;x;x-=lowbit(x)) res+=c[x];
return res;
}
void solve(int l,int r,int ql,int qr)
{
if(l>r||ql>qr) return;
if(l==r)
{
for(int i=ql;i<=qr;i++)
if(q[i].type) ans[q[i].pos]=l;
return;
}
int cnt1=0,cnt2=0;
int mid=(l+r)>>1;
for(int i=ql;i<=qr;i++)
if(q[i].type)
{
int tmp=sum(q[i].y)-sum(q[i].x-1);
if(q[i].k<=tmp) q1[++cnt1]=q[i];
else q[i].k-=tmp,q2[++cnt2]=q[i];
}
else
{
if(q[i].x<=mid) add(q[i].pos,q[i].k),q1[++cnt1]=q[i];
else q2[++cnt2]=q[i];
}
for(int i=1;i<=cnt1;i++)
if(q1[i].type==0) add(q1[i].pos,-q1[i].k);
for(int i=1;i<=cnt1;i++) q[ql+i-1]=q1[i];
for(int i=1;i<=cnt2;i++) q[ql+cnt1+i-1]=q2[i];
solve(l,mid,ql,ql+cnt1-1);
solve(mid+1,r,ql+cnt1,qr);
}
int a[N];
int main()
{
n=read(),m=read();
for(int i=1;i<=n;i++)
{
a[i]=read();
q[++cnt]=Query(a[i],0,1,i,0);
}
int tot=0;
for(int i=1;i<=m;i++)
{
char ch=getchar();while(ch!='Q'&&ch!='C')ch=getchar();
int x=read(),y=read(),z;
if(ch=='C') q[++cnt]=Query(a[x],0,-1,x,0),q[++cnt]=Query(a[x]=y,0,1,x,0);
else z=read(),q[++cnt]=Query(x,y,z,++tot,1);
}
solve(-INF,INF,1,cnt);
for(int i=1;i<=tot;i++)
write(ans[i]);
return 0;
}
和树状数组套主席树的对比:
整体二分:
跑了(2.33s) 【滑稽】。
树状数组套主席树:
(15.61s),空间和时间都被整体二分吊起来锤。。。
可见整体二分的优势。。。
P3332 [ZJOI2013]K大数查询
一道毒瘤题。。。(把题号倒过来)
题意有点鬼畜。。。建议看一下样例说明。。。
注意是第(k)大不是第(k)小。。。
点修改变成了区间修改。。。怎么办?
把树状数组换成线段树就好了。。。
注意比(mid)小的数的个数会爆(int)。。。 害我调了一上午
#include <bits/stdc++.h>
using namespace std;
#define debug(...) fprintf(stderr,__VA_ARGS__)
typedef long long ll;
const int MAXN=200010;
const ll INF=2e18;
struct seg{
int l,r;
ll add,sum;
}t[MAXN<<2];
void pushup(int x){
t[x].sum=t[x<<1].sum+t[x<<1|1].sum;
}
void pushdown(int x){
if (!t[x].add) return;
int l=t[x].l,r=t[x].r,mid=(l+r)>>1;
t[x<<1].add+=t[x].add;
t[x<<1|1].add+=t[x].add;
t[x<<1].sum+=t[x].add*(mid-l+1);
t[x<<1|1].sum+=t[x].add*(r-mid);
t[x].add=0;
}
void build(int x,int l,int r){
t[x]=(seg){l,r,0,0};
if (l==r)
return;
int mid=(l+r)>>1;
build(x<<1,l,mid);
build(x<<1|1,mid+1,r);
pushup(x);
}
void update(int x,int ql,int qr,ll v){
int l=t[x].l,r=t[x].r;
if (ql<=l&&r<=qr){
t[x].add+=v;
t[x].sum+=v*(r-l+1);
return;
}
pushdown(x);
int mid=(l+r)>>1;
if (ql<=mid) update(x<<1,ql,qr,v);
if (mid<qr) update(x<<1|1,ql,qr,v);
pushup(x);
}
ll query(int x,int ql,int qr){
int l=t[x].l,r=t[x].r;
if (ql<=l&&r<=qr) return t[x].sum;
pushdown(x);
int mid=(l+r)>>1;ll res=0;
if (ql<=mid) res+=query(x<<1,ql,qr);
if (mid<qr) res+=query(x<<1|1,ql,qr);
return res;
}
ll ans[MAXN];
int n,m;
struct event{
int opt,x,y;ll v;int id;
void print(){
debug("%d %d %d %lld
",opt,x,y,v);
}
}q[MAXN],q1[MAXN],q2[MAXN];
void solve(ll l,ll r,int ql,int qr){
if (ql>qr||l>r) return;
if (l==r){
for (int i=ql;i<=qr;i++)
if (q[i].opt==2) ans[q[i].id]=l;
return;
}
ll mid=(l+r)>>1;
int cnt1=0,cnt2=0;
for (int i=ql;i<=qr;i++){
if (q[i].opt==1){
if (q[i].v>mid){
update(1,q[i].x,q[i].y,1);
q1[++cnt1]=q[i];
}else
q2[++cnt2]=q[i];
}else{
ll tmp=query(1,q[i].x,q[i].y);
if (tmp>=q[i].v)
q1[++cnt1]=q[i];
else{
q[i].v-=tmp;
q2[++cnt2]=q[i];
}
}
}
for (int i=1;i<=cnt1;i++)
if (q1[i].opt==1&&q1[i].v>mid) update(1,q1[i].x,q1[i].y,-1);
for (int i=ql;i<ql+cnt1;i++)
q[i]=q1[i-ql+1];
for (int i=ql+cnt1;i<=qr;i++)
q[i]=q2[i-ql-cnt1+1];
solve(mid+1,r,ql,ql+cnt1-1);
solve(l,mid,ql+cnt1,qr);
}
int main(){
scanf("%d%d",&n,&m);
build(1,1,n);
int tot=0;
for (int i=1;i<=m;i++){
int opt,a,b;ll c;
scanf("%d%d%d%lld",&opt,&a,&b,&c);
q[i]=(event){opt,a,b,c,opt==2?++tot:0};
}
solve(-n,n,1,m);
for (int i=1;i<=tot;i++)
printf("%lld
",ans[i]);
return 0;
}
好像有线段树套平衡树的做法。。。但复杂度就变成了三个(log)了。。。(二分一个(log),树套树的单个查询(2)个),没写。。。不会
从上面三个例题来看,整体二分有以下几个优势:
-
时空复杂度都比树套树优越;
-
对比树套树代码的复杂度简单一点;
-
蒟蒻专属
但它好像只支持离线。。。这个就自求多福吧(qwq)。。。
整体二分大法吼啊!
PS:整体二分的复杂度(O(nlog^2n)),就看做是(O(log)值域大小(*logn*n))吧。。。