LibreOJ-109 并查集

题目描述

这是一道模板题。

维护一个  点的无向图，支持：

• 加入一条连接  和  的无向边
• 查询  和  的连通性

由于本题数据较大，因此输出的时候采用特殊的输出方式：用  或  代表每个询问的答案，将每个询问的答案依次从左到右排列，把得到的串视为一个二进制数，输出这个二进制数  的值。

请务必使用快读。

输入格式

第一行包含两个整数 ，表示点的个数和操作的数目。

接下来  行每行包括三个整数 。

• 如果 ，则表示加入一条连接  和  的无向边；
• 如果 ，则表示查询  和  的连通性。

输出格式

一行包括一个整数表示答案。

样例

输入复制

3 6
1 1 0
0 0 1
1 0 1
1 1 2
0 2 1
1 2 1

输出复制

5

答案串为 。

数据范围与提示

n<=4000000,m<=8000000

By zyz

 

 

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
int p[4000010];
long long ans=0;
int mod=998244353;
int n,m;
int a,b,c;

int find(int x)
{
	if(p[x]!=x)
	return p[x]=find(p[x]);
	return p[x];
}
void merge(int x,int y)//void
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	{
		p[fx]=fy;
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
	p[i]=i;//初始化 
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&a,&b,&c);
		if(a==0)//表示把b，c连通 
		{
			merge(b,c);
		}
		else// 查询b，c的连通性 
		{
			if(find(b)==find(c))
			{
				ans=ans*2;
				ans++;
				ans=ans%mod;
			}
			else
			{
				ans=ans*2;
				ans=ans%mod;
			}
		}
	}
	printf("%d\n",ans%mod);
	return 0;
}