线段树分治-学习笔记

线段树分治-学习笔记

阅前须知：本文给出了线段树分治的一道例题以及多道习题，同时给出了部分实现的代码，帮助学习线段树分治。

总述

线段树分治是一种离线算法，在于把修改挂在线段树的节点上，通过遍历线段树求出每个叶子节点的答案，以减小复杂度。

例题 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);
}