bzoj3237 [Ahoi2013]连通图 线段树分治+并查集按秩合并

Description

---

给定n个点m条边的无向图，k次询问，每次删除s条边并询问此时图的连通性，询问互相独立。
n<=1e5,m<=2e5,k<=1e5,s<=4

Solution

---

传说中的线段树分治
删除和插入同时存在的话非常麻烦，因此考虑一种处理询问的顺序使得只剩插入操作
我们把一条边存在的询问区间扔进线段树，然后遍历整棵线段树。遍历的时候就插入区间内的边，同时压进一个栈里。回溯的时候按照插入顺序的逆序删除并弹栈。这样在叶子处判断就可以了。写起来就像另一个版本的cdq

判断图连通可以并查集维护集合size，要求资瓷可撤销可以可持久化也可以按秩合并，高兴就好

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
#include <stack>
#define rep(i,st,ed) for (register int i=st;i<=ed;++i)
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)<(y)?(x):(y))

const int N=200005;

struct edge {int x,y;} ;

int fa[N],size[N],d[N],pre[N],h[N],top,n;

std:: vector <int> vec[N<<2];

edge e[N],stack[N];

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return x*v;
}

int find(int x) {
	return (!fa[x])?(x):(find(fa[x]));
}

void ins(int now,int tl,int tr,int l,int r,int x) {
	if (r<l) return ;
	if (tl>=l&&tr<=r) {
		vec[now].push_back(x);
		return ;
	}
	int mid=(tl+tr)>>1;
	ins(now<<1,tl,mid,l,min(r,mid),x);
	ins(now<<1|1,mid+1,tr,max(mid+1,l),r,x);
}

void solve(int now,int l,int r) {
	int wjp=top;
	for (int i=0,_=vec[now].size();i<_;++i) {
		edge ed=e[vec[now][i]];
		int x=find(ed.x),y=find(ed.y);
		if (x==y) continue;
		if (h[x]<h[y]) fa[x]=y,size[y]+=size[x],stack[++top]=(edge) {x,0};
		else if (h[x]>h[y]) fa[y]=x,size[x]+=size[y],stack[++top]=(edge) {y,0};
		else if (h[x]==h[y]) {
			fa[y]=x,size[x]+=size[y],h[x]++,stack[++top]=(edge) {y,1};
		}
	}
	if (l==r) {
		puts((size[find(1)]==n)?"Connected":"Disconnected");
	} else {
		int mid=(l+r)>>1;
		solve(now<<1,l,mid);
		solve(now<<1|1,mid+1,r);
	}
	for (;top>wjp;) {
		edge lxf=stack[top--];
		int ff=fa[lxf.x],x=lxf.x; fa[x]=0;
		size[ff]-=size[x];
		if (lxf.y) h[ff]--;
	}
}

int main(void) {
	freopen("data.in","r",stdin);
	freopen("myp.out","w",stdout);
	n=read(); int m=read();
	rep(i,1,n) size[i]=1;
	rep(i,1,m) {
		e[i].x=read(),e[i].y=read();
		pre[i]=1;
	}
	int T=read();
	rep(i,1,T) {
		for (int k=read();k--;) {
			int x=read();
			if (pre[x]<i) ins(1,1,T,pre[x],i-1,x);
			pre[x]=i+1;
		}
	}
	rep(i,1,m) if (pre[i]<=T) ins(1,1,T,pre[i],T,i);
	solve(1,1,T);
	return 0;
}