P1653 猴子 [并查集]

传送门

考虑暴力 , 我们可以一次一次短边 , 然后判断连通性 , 显然是O(mn)的

对于不容易判断删除后的连通性的 , 我们转换为倒序插入

我们倒着插入 , 然后记录当前点与1连通的时刻

发现一个点的答案就是它或它的祖先与1连通的时间的最小值 , find的时候更新一下即可

#include<bits/stdc++.h>
#define N 400050
using namespace std;
int n,m,ch[N][2],q[N][2];
int fa[N],ans[N],flag[N][2];
int find(int x){
	if(x==fa[x]) return x;
	int rt=find(fa[x]);
	ans[x] = min(ans[x],ans[fa[x]]);
	return fa[x]=rt;
}
int Find(int x){
	if(x==fa[x]) return x;
	int rt=Find(fa[x]);
	ans[x] = min(ans[x],ans[fa[x]]);
	return fa[x]=rt;
}
void Merge(int x,int y,int val){
	x=find(x),y=find(y);
	if(x!=y){
		if(x==1) fa[y]=x , ans[y]=val;
		else fa[x]=y , ans[x]=val;
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d%d",&ch[i][0],&ch[i][1]);
		fa[i]=i;
	}
	for(int i=1;i<=m;i++){
		 scanf("%d%d",&q[i][0],&q[i][1]);
		 flag[q[i][0]][--q[i][1]] = 1;
	}
	for(int i=1;i<=n;i++){
		if(!flag[i][0] && ch[i][0]!=-1) Merge(ch[i][0],i,0);
		if(!flag[i][1] && ch[i][1]!=-1) Merge(ch[i][1],i,0);
	}
	for(int i=1;i<=n;i++) ans[i]=m+1;
	for(int i=m;i>=1;i--){
		int x=q[i][0],y=ch[q[i][0]][q[i][1]];
		if(y!=-1) Merge(x,y,i);
	}
	for(int i=1;i<=n;i++){
		find(i); printf("%d\n",ans[i]>m?-1:ans[i]-1);
	} return 0;
}

猜你喜欢

转载自blog.csdn.net/sslz_fsy/article/details/85040291