可撤销并查集板子 WA88pts 求调
查看原帖
可撤销并查集板子 WA88pts 求调
564732
TimSwn090306楼主2024/2/29 18:52

rt,第一次写可撤销并查集,感觉没啥问题,但是 WA88pts ,求大佬帮调qwq

提交记录

代码:

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
const int maxm=2e5+5;
struct edge{
	int to,next;
}e[maxm];
struct query{
	int opt,x,y;
}a[maxm];
struct dsu{
	int top,fa[maxn],siz[maxn],stk[maxn];
	inline void init(int x){
		top=0;
		for (int i=1;i<=x;i++) fa[i]=i,siz[i]=1;
	}
	inline int find(int x){
		if (fa[x]==x) return x;
		return find(fa[x]);
	}
	inline void merge(int x,int y){
		x=find(x),y=find(y);
		if (x==y) return ;
		if (siz[x]<siz[y]) swap(x,y);
		fa[y]=x,siz[x]+=siz[y];
		stk[++top]=y;
	}
	inline void back(){
		if (!top) return ;
		siz[fa[stk[top]]]-=siz[stk[top]];
		fa[stk[top]]=stk[top];
		top--;
	}
}f;
int n,m,tot,h[maxm],ans[maxm];
inline void addEdge(int x,int y){
	e[++tot]=(edge){y,h[x]};
	h[x]=tot;
}
inline void dfs(int x){
	for (int i=h[x];i;i=e[i].next){
		int v=e[i].to;
		if (a[v].opt==3){
			ans[v]=(f.find(a[v].x)==f.find(a[v].y));
			dfs(v);
		}else if (a[v].opt==1){
			f.merge(a[v].x,a[v].y);
			dfs(v);
			f.back();
		}else dfs(v);
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for (int i=1;i<=m;i++){
		scanf("%d",&a[i].opt);
		if (a[i].opt==2){
			scanf("%d",&a[i].x);
			addEdge(a[i].x,i);
		}else{
			scanf("%d%d",&a[i].x,&a[i].y);
			addEdge(i-1,i);
		}
	}
	f.init(n);
	dfs(0);
	for (int i=1;i<=m;i++) if (a[i].opt==3) printf("%d\n",ans[i]);
	return 0;
}
2024/2/29 18:52
加载中...