线段树分治-学习笔记
线段树分治-学习笔记
阅前须知：本文给出了线段树分治的一道例题以及多道习题，同时给出了部分实现的代码，帮助学习线段树分治。
总述
线段树分治是一种离线算法，在于把修改挂在线段树的节点上，通过遍历线段树求出每个叶子节点的答案，以减小复杂度。
例题 P5787 二分图
题目大意：\(n\) 个点的图上，有 \(m\) 条边，第 \(i\) 条边 \((x,y)\) 在时刻 \(l_i\sim r_i\) 存在，问每个时刻的图是否是二分图。
解题思路：首先如果是动态加边，没有删边，怎么判断一个图是否是二分图？
可以考虑种类并查集，一共两类，一条边连接不同种类，如果连边时发现两个点属于同一种类，那么就不是二分图。这样我们就能做到 \(O(mk\alpha(n))\)。
考虑分治思想，我们把不同的询问取出它们的共同的修改，这样就能减小复杂度。
考虑把时刻 \(l_i\sim r_i\) 的修改挂在线段树上的 \(\log k\) 个区间上，这样一共有 \(m\log k\) 个修改，是可接受的。接下来我们从树根开始走，走到一个点时把修改加上，离开这个子树时把修改撤回，走到叶子时就可以记录答案了。
我们可以使用启发式合并的并查集，这样在 getfa 时就不用路径压缩，然后合并时用 vector 记录 fa 和 sz 的每个修改的位置与原来的值，在回溯时把这些修改撤销即可。
时间复杂度 \(O(k\log k\log n)\)。
参考实现：
int fa[N],sz[N];
int getfa(int x) {
if(fa[x]==x) return x;
return getfa(fa[x]);
}
#define mid ((l+r)>>1)
#define ls (x<<1)
#define rs (ls|1)
vp up[N];
void add(int x,int l,int r,int L,int R,int a,int b) {
if(R<l||r<L)return;
if(L<=l&&r<=R) {
up[x].pb({a,b});
return;
}
add(ls,l,mid,L,R,a,b),add(rs,mid+1,r,L,R,a,b);
}
void solve(int x,int l,int r,int flag) {
vp o1,o2;
if(!flag)for(auto [a,b]:up[x]) {
int u=getfa(a),v=getfa(b);
if(u==v) flag=1;
u=getfa(a),v=getfa(b+n);
if(u!=v) {
if(sz[u]<sz[v]) swap(u,v);
o1.pb({u,sz[u]}),o2.pb({v,fa[v]});
sz[u]+=sz[v],fa[v]=u;
}
u=getfa(a+n),v=getfa(b);
if(u!=v) {
if(sz[u]<sz[v]) swap(u,v);
o1.pb({u,sz[u]}),o2.pb({v,fa[v]});
sz[u]+=sz[v],fa[v]=u;
}
}
if(l==r) {
if(flag) write("No\n");
else write("Yes\n");
for(auto [a,b]:o1) sz[a]=b;
for(auto [a,b]:o2) fa[a]=b;
return;
}
solve(ls,l,mid,flag);
solve(rs,mid+1,r,flag);
reverse(o1.begin(),o1.end());
reverse(o2.begin(),o2.end());
for(auto [a,b]:o1) sz[a]=b;
for(auto [a,b]:o2) fa[a]=b;
}
signed main(){
read(n,m,K);
fo(i,1,n*2) fa[i]=i,sz[i]=1;
fo(i,1,m) {
int l,r,x,y;
read(x,y,l,r);
++l,++r;
if(l==r) continue;
add(1,1,K,l,r-1,x,y);
}
solve(1,1,K,0);
return 0;
}
练习A AHOI2013 连通图
题目大意：每次询问断掉若干条边，问图是否还连通。
解题思路：我们把每条边断开的位置找出来，这样就分成了若干个区间，每个区间都表示这条边连上。
考虑线段树分治，把区间挂在线段树上。那么考虑怎么判断是否连通，用并查集缩点，如果成功缩点连通块个数减一，判断连通块个数是否为 1 即可。
参考实现：
const int N=1e6+5;
int X[N],Y[N];
vi d[N];
int fa[N],sz[N];
int getfa(int x) {
if(fa[x]==x)return x;
return getfa(fa[x]);
}
int n,m,K;
#define ls (x<<1)
#define rs (ls|1)
#define mid ((l+r)>>1)
vp c[N];
void add(int x,int l,int r,int L,int R,int a,int b) {
if(R<l||r<L) return;
if(L<=l&&r<=R) {
c[x].pb({a,b});
return;
}
add(ls,l,mid,L,R,a,b),add(rs,mid+1,r,L,R,a,b);
}
void solve(int x,int l,int r,int s) {
vp o1,o2;
for(auto [a,b]:c[x]) {
a=getfa(a),b=getfa(b);
if(a!=b) {
if(sz[a]<sz[b]) swap(a,b);
o1.pb({a,sz[a]}),o2.pb({b,fa[b]});
fa[b]=a,sz[a]+=sz[b];
--s;
}
}
if(l==r) {
if(s==1) write("Connected\n");
else write("Disconnected\n");
reverse(o1.begin(),o1.end());
reverse(o2.begin(),o2.end());
for(auto [a,b]:o1) sz[a]=b;
for(auto [a,b]:o2) fa[a]=b;
return;
}
solve(ls,l,mid,s);
solve(rs,mid+1,r,s);
reverse(o1.begin(),o1.end());
reverse(o2.begin(),o2.end());
for(auto [a,b]:o1) sz[a]=b;
for(auto [a,b]:o2) fa[a]=b;
}
signed main(){
read(n,m);
fo(i,1,m) read(X[i],Y[i]);
read(K);
fo(i,1,K) {
int t; read(t);
fo(j,1,t) {
int x; read(x);
d[x].pb(i);
}
}
fo(i,1,m) {
int lst=0;
for(auto j:d[i]) {
if(lst+1<j) add(1,1,K,lst+1,j-1,X[i],Y[i]);
lst=j;
}
if(lst<K) add(1,1,K,lst+1,K,X[i],Y[i]);
}
fo(i,1,n) sz[i]=1,fa[i]=i;
solve(1,1,K,n);
return 0;
}
练习B CF813F
记录每一种边上一次出现的位置即可转化为例题。
参考实现：
const int N=1e6+5;
int fa[N],sz[N];
int getfa(int x) {
if(fa[x]==x) return x;
return getfa(fa[x]);
}
int tot;
map<pi,int> mp;
int X[N],Y[N];
int lst[N],opt[N];
vp d[N];
#define ls (x<<1)
#define rs (ls|1)
#define mid ((l+r)>>1)
void add(int x,int l,int r,int L,int R,int a,int b) {
if(L<=l&&r<=R) {
d[x].pb({a,b});
return;
}
if(R<l||r<L) return;
add(ls,l,mid,L,R,a,b),add(rs,mid+1,r,L,R,a,b);
}
void solve(int x,int l,int r,int flag){
vp o1,o2;
for(auto [a,b]:d[x]) {
int u=getfa(a),v=getfa(b);
if(u==v) flag=1;
v=getfa(b+n);
if(u!=v) {
if(sz[u]<sz[v]) swap(u,v);
o1.pb({u,sz[u]}),o2.pb({v,fa[v]});
sz[u]+=sz[v],fa[v]=u;
}
u=getfa(a+n),v=getfa(b);
if(u!=v) {
if(sz[u]<sz[v]) swap(u,v);
o1.pb({u,sz[u]}),o2.pb({v,fa[v]});
sz[u]+=sz[v],fa[v]=u;
}
}
if(l==r) {
if(flag) write("NO\n");
else write("YES\n");
reverse(o1.begin(),o1.end());
reverse(o2.begin(),o2.end());
for(auto [a,b]:o1) sz[a]=b;
for(auto [a,b]:o2) fa[a]=b;
return;
}
solve(ls,l,mid,flag);
solve(rs,mid+1,r,flag);
reverse(o1.begin(),o1.end());
reverse(o2.begin(),o2.end());
for(auto [a,b]:o1) sz[a]=b;
for(auto [a,b]:o2) fa[a]=b;
}
signed main(){
read(n,Q);
fo(i,1,2*n) fa[i]=i,sz[i]=1;
fo(i,1,Q) {
int x,y; read(x,y);
if(mp.find({x,y})==mp.end())mp[{x,y}]=++tot,X[tot]=x,Y[tot]=y;
int t=mp[{x,y}];
opt[t]^=1;
if(opt[t]==0) {
add(1,1,Q,lst[t],i-1,x,y);
}
lst[t]=i;
}
fo(i,1,tot) if(opt[i]) add(1,1,Q,lst[i],Q,X[i],Y[i]);
solve(1,1,Q,0);
}
浙公网安备 33010602011771号