易爆物 LA 3644【并查集】

水题

并查集维护是否有环。如果有两个节点在同一个集合里面。再添加这两个节点会形成环

#include <bits/stdc++.h>
using namespace std;
const int maxn =1e5+50;
typedef long long ll;
int fa[maxn];
void init()
{
	for(int i=0;i<maxn;i++) fa[i]=i;
}
int findroot(int u)
{
	if(fa[u]==u) return u;
	else return fa[u] = findroot(fa[u]);
}
bool same(int u,int v)
{if(findroot(u)==findroot(v)) return true;else return false;}
void Union(int u,int v)
{
	u = findroot(u);
	v=findroot(v);
	if(u==v) return;
	else fa[u]=v;
}
int ans=0;
int main()
{
	int x,y;
	init();
	while(cin>>x)
	{
		if(x==-1)
		{
			printf("%d\n",ans);
			ans=0;
			init();
		}
		else
		{
			cin>>y;
			if(same(x,y))ans++;
			else Union(x,y);
		}

	}
	return 0;
}
全部评论

相关推荐

11-28 17:48
中山大学 C++
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务