AcWing 238. 银河英雄传说 并查集模板题

Posted 2021-12-02 karshey

题
参考

#include<bits/stdc++.h>
using namespace std;
#define fir(i,a,n) for(int i=a;i<=n;i++)
//======================
const int N=3e4+10;
int fa[N],cnt[N],dis[N];//dis[i]表示第i个到头的距离

int find(int x)//找根的同时维护dis数组 
{
	if(fa[x]!=x)
	{
		int root=find(fa[x]);//找到fa的根 
		dis[x]+=dis[fa[x]];//！！！ 
		fa[x]=root;
	}
	return fa[x];
} 

void merge(int a,int b)//a接在b后面 a的dis要加pb的cnt pb的cnt要加a的cnt 
{
	int pa=find(a),pb=find(b);
	dis[pa]+=cnt[pb];
	cnt[pb]+=cnt[pa];
	fa[pa]=pb;
}
int main()
{
	int t;scanf("%d",&t);
	fir(i,0,3e4) 
	{
		fa[i]=i;
		cnt[i]=1;
	}
	while(t--)
	{
		string ch;int a,b;
		cin>>ch>>a>>b;
		if(ch=="M")
		{
			//a接在b后面 
			merge(a,b);
		}
		else
		{
			int pa=find(a),pb=find(b);
			if(pa!=pb) 
			{
				printf("-1\\n");continue;
			}
			else printf("%d\\n",max(0,abs(dis[a]-dis[b])-1));			
		}
	}
	return 0; 
}