2019.01.22 uoj#14. 【UER #1】DZY Loves Graph(并查集)

时间:2022-10-11 21:47:48

传送门

题意简述:

要求支持以下操作:

在a与b之间连一条长度为i的边(i是操作编号);删除当前图中边权最大的k条边;表示撤销第 i−1次操作,保证第1次,第i−1 次不是撤回操作。

要求在每次操作后输出当前图的最小生成树边权和。


思路:由于边权为当前操作编号因此相当于边是单调加入的,也就是说我们可以直接上kruskalkruskalkruskal的合并方法。

关键在于怎么维护这几个操作。

加边操作:加入一条边。

删除操作:删掉最近加入的kkk条边。

撤回加边操作:删掉最近加入的一条边

撤回删除操作:等于没有变化。

那么我们用栈存下每一个状态的值转移给后面的状态即可。

注意并查集要简单可持久化一下,可以用按秩合并。

实测了一波按sizesizesize合并比按深度合并快。

按sizesizesize合并代码:

#include<bits/stdc++.h>
#define ri register int
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
typedef long long ll;
const int N=5e5+5;
int n,m,fa[N],siz[N],cnt[N],stk[N],top=0;
ll ans[N];
inline int find(int x){return x==fa[x]?x:find(fa[x]);}
inline void print(int x){cout<<(cnt[x]==n-1?ans[x]:0)<<'\n';}
inline void add(int x,int y,int w){
	int fx=find(x),fy=find(y);
	++top,cnt[top]=cnt[top-1],ans[top]=ans[top-1];
	if(fx==fy)stk[top]=0;
	else{
		if(siz[fx]<siz[fy])swap(fx,fy);
		stk[top]=fy,fa[fy]=fx,siz[fx]+=siz[fy],++cnt[top],ans[top]+=w;
	}
}
inline void delet(int tim){
	while(tim--){
		int p=stk[top--];
		siz[fa[p]]-=siz[p],fa[p]=p;
	}
}
int main(){
	n=read(),m=read();
	for(ri i=1;i<=n;++i)fa[i]=i,siz[i]=1;
	for(ri last=0,k,i=1;i<=m;++i){
		char s[6];
		scanf("%s",s);
		if(s[0]=='A'){
			if(last==2)delet(k);
			int a=read(),b=read();
			add(a,b,i),print(top),last=1;
		}
		else if(s[0]=='D'){
			if(last==2)delet(k);
			int a=read();
			print(top-a),last=2,k=a;
		}
		else{
			if(last==1)delet(1);
			print(top),last=0;
		}
	}
	return 0;
}

按深度合并代码:

#include<bits/stdc++.h>
#define ri register int
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
typedef long long ll;
const int N=3e5+5,M=5e5+5;
int n,m,fa[N],rk[N],cnt[M],stk[M],top=0;
ll ans[M];
inline int find(int x){return x==fa[x]?x:find(fa[x]);}
inline void print(int x){cout<<(cnt[x]==n-1?ans[x]:0)<<'\n';}
inline void add(int x,int y,int w){
	int fx=find(x),fy=find(y);
	++top,cnt[top]=cnt[top-1],ans[top]=ans[top-1];
	if(fx==fy)stk[top]=0;
	else{
		if(rk[fx]<rk[fy])swap(fx,fy);
		stk[top]=fy,fa[fy]=fx,rk[fx]+=rk[fx]==rk[fy],++cnt[top],ans[top]+=w;
	}
}
inline void delet(int tim){
	while(tim--){
		int p=stk[top--];
		rk[fa[p]]-=rk[fa[p]]==rk[p]+1,fa[p]=p;
	}
}
int main(){
	n=read(),m=read();
	for(ri i=1;i<=n;++i)fa[i]=i,rk[i]=1;
	for(ri last=0,k,i=1;i<=m;++i){
		char s[6];
		scanf("%s",s);
		if(s[0]=='A'){
			if(last==2)delet(k);
			int a=read(),b=read();
			add(a,b,i),print(top),last=1;
		}
		else if(s[0]=='D'){
			if(last==2)delet(k);
			int a=read();
			print(top-a),last=2,k=a;
		}
		else{
			if(last==1)delet(1);
			print(top),last=0;
		}
	}
	return 0;
}