程序自动分析(并查集+排序)

题意 给许多个x,y,k,若k=1,xy,否则x!=y,如果矛盾,输出NO,否则YES
对于k=1,并查集简单操作一下,k=0,如果find(x)
find(y),打个标记,输出NO;
有一个需要注意的地方是,对于询问我们要进行sort,使k=1的情况先执行,这样可以保证最后判断的答案正确。

#include<iostream>
#include<cstdio>
using namespace std;
int re(){
	char c=getchar();int all=0,pd=1;
	for(;c>'9'||c<'0';c=getchar()) if(c=='-') pd=-1;
	while(c>='0'&&c<='9') all=all*10+c-'0',c=getchar();return all*pd;
}const int N=3e4+5;int f[N],front[N],num[N];
int find(int x){
	if(x==f[x]) return x;
	int fa=find(f[x]);front[x]+=front[f[x]];return f[x]=fa;
}int abs(int x){return x>0?x:-x;}
int main(){
	int n=re();for(int i=1;i<=30000;i++) num[i]=1,f[i]=i;
	for(int i=1;i<=n;i++){
		char c;cin>>c;
		int x=re(),y=re();
		int r1=find(x),r2=find(y);
		if(c=='M'){
			front[r1]+=num[r2];
			f[r1]=r2;num[r2]+=num[r1];
			num[r1]=0;
		}else if(r1!=r2) printf("-1\n");
		else printf("%d\n",abs(front[x]-front[y])-1);
	}
}
posted @ 2018-08-22 20:51  BLUE_EYE  阅读(311)  评论(0编辑  收藏  举报