专题训练3—Aproblem 并查集

题意

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

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

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

输入格式

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

接下来 mm 行每行包括三个整数 op,u,v。

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

输出格式

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

思路

并查集的板题，就是先让所有点的根都是自己接着就是并根让两个连通的根一样，最后就是查。

代码

#include<bits/stdc++.h>
using namespace std;

int father[4000010];
long long ans=0;
int p=0,temp;
const long long mod=998244353;


int read() 
{
    int x=0,f=1;
    char c=getchar();
    while(c<'0'||c>'9'){if(c=='-') f=-1;c=getchar();}
    while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x*f;
}


int find(int k)
{
	if(father[k]!=k) 
		father[k]=find(father[k]);
	return father[k];
}

void _union(int x,int y)
{
	if(x>y) temp=x,x=y,y=temp;
	x=find(x);
	y=find(y);
	if(x!=y)
		father[y]=x;
} 

int main()
{
	int op,u,v,n,m;n=read();m=read();
	for(int i=1;i<=n;i++)
		father[i]=i;
	while(m--)
	{
		op=read();u=read();v=read();
		if(op==0)
			_union(u,v);
		else if(op==1)
		{
			if(find(u)==find(v))
			{
				ans*=2;
				ans+=1;
				ans%=mod;
				p++;
			}
			else if(p!=0)
			{
			 	ans*=2;
			 	ans%=mod;
			}
		}
	}
	cout<<ans;
	return 0;
}