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

karshey

于 2021-10-29 19:22:49 发布

阅读量142

点赞数

分类专栏：数据结构题我的ACM之路文章标签：并查集 C++

本文链接：https://blog.youkuaiyun.com/karshey/article/details/121041174

版权

我的ACM之路同时被 2 个专栏收录

159 篇文章

订阅专栏

数据结构题

3 篇文章

订阅专栏

本文探讨了一种使用C++编写的高效算法，通过find和merge函数处理图中节点间的连接与距离计算。主要关注于并查集数据结构的应用，以及如何在`merge`操作中更新`dis`数组以反映节点间的关系。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题
 参考

#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; 
}