NE的挑战II 带权并查集

有隐藏仇恨结构体数组

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+5;
int n,m,pre[maxn],cnt[maxn],cntt;
bool vis[maxn];	//隐藏仇恨
struct node{
	int p1;
	int p2;
}h[maxn];
void init(){
	for(int i=1;i<=n;i++){
		pre[i]=i+n;
		pre[i+n]=i+n;
		cnt[i+n]=1;
		cnt[i]=0;//原先的123盒子清0 
	}
}
int find(int x){
	return pre[x]==x?x:pre[x]=find(pre[x]);
}
void Union(int x,int y){
	int fx=find(x);
	int fy=find(y);
	if(fx==fy)
		return;
	cnt[fx]+=cnt[fy];
	pre[fy]=fx;
	
}
void SpeUnion(int x,int y){
	int fx=find(x);
	int fy=find(y);
	cnt[fx]--;
	cnt[fy]++;
	pre[x]=fy;
}
int main(){
	cin>>n>>m;
	init();
	int a,b,c;
	while(m--){
		cin>>a;
		if(a==1){
			cin>>b>>c;
			Union(b,c);
		}
		else if(a==2){
			cin>>b>>c;
			SpeUnion(b,c);
		}
		else if(a==3){
			cin>>b;
			int fb=find(b);
			cout<<cnt[fb]-1<<endl;
		}
		else if(a==4){
			cin>>b>>c;
			int fb=find(b);
			int fc=find(c);
			if(fb==fc)
				cout<<"Yes"<<endl;
			else
				cout<<"No"<<endl;
		}
		else if(a==5){
			cin>>b>>c;
			h[cntt].p1=b;
			h[cntt].p2=c;
			cntt++;
		}
	}
	for(int i=0;i<cntt;i++){
		int fx=find(h[i].p1);
		int fy=find(h[i].p2);
		if(fx==fy){
			vis[fx]=1;
		}
	}
	int ans=-1;	//不存在无隐藏仇恨的盟军 
	for(int i=1;i<=n;i++){
		int fx=find(i);
		if(vis[fx])
			continue;
		ans=max(ans,cnt[fx]);
	}
	cout<<ans<<endl;
	return 0;
}