LibreOJ - 109 并查集

https://vjudge.net/problem/LibreOJ-109

#include<iostream>
#include<algorithm>
using namespace std;
const int N=4e6+10;
int n,m,mod=998244353;
int p[N],q[N];
void read(int &x)
{
	x=0; bool f=false;
	char c=getchar();
	while(!isdigit(c))
	{
		if(c=='-')
			f=1;
		c=getchar();
	}
	while(isdigit(c))
	{
		x=x*10+c-'0';
		c=getchar();
	}
	x=f? (~x)+1:x;
}
int find(int x)
{
	if(p[x]!=x) p[x]=find(p[x]);
	return p[x];
}
int main()
{
	read(n);
	read(m);
	for(int i=0;i<=n;i++) p[i]=i;
	int f,a,b,k=0;
	while(m--)
	{
		read(f);
		read(a);
		read(b);
		a=find(a),b=find(b);
		if(!f)
			p[a]=b;
		else
		{
			if(a==b)
				q[k]=1;
			else
				q[k]=0;
			k++;
		}
	}
	long long res=0;
	for(int i=0;i<k;i++)
		res=(res*2+q[i])%mod;
	//快速幂的处理也可以合并到else里边,,每次查询的同时计算res
	cout<<res<<"\n";
	return 0;
}