其他分享
首页 > 其他分享> > ZR 动物园

ZR 动物园

作者:互联网

30分做法:

三进制枚举,暴力模拟即可。

#include<queue>
#include<cstring>
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
const ll mo=998244353;
const int maxm=2e5+7;
int n,m;
int f[maxm];
ll w[maxm];
ll tot,inv3;
ll qpow(int a,int b)
{
 ll ans=1,base=a;
 while(b)
 {
  if(b&1)
  ans=ans*base%mo;
  base=base*base%mo;
  b>>=1;	
 }
 return ans;
}
int find(int x)
{
 if(x!=f[x])
 {
  int father=find(f[x]);
  if(father!=f[x])
  w[x]=w[x]*w[f[x]]%mo;//先不处理根节点的值,防止合并时再计算一次
  f[x]=father;
 }
 return f[x];
}
int main()
{
 scanf("%d%d",&n,&m);
 for(int i=1;i<=n;i++)
 {
  f[i]=i;
  w[i]=1;	
 }
 tot=qpow(3,n);
 inv3=qpow(3,mo-2);//1/3在%mo下的乘法逆元 
 for(int i=1;i<=m;i++)
 {
  int opt,u,v;
  scanf("%d%d",&opt,&u);
  if(opt==1)
  {
    scanf("%d",&v);
	int fx=find(u);
	int fy=find(v);//因为使过的不能使,所以不用判fx,fy相等
	f[fy]=fx;
	w[fx]=w[fx]*inv3*2ll%mo;
	w[fy]=w[fy]*inv3%mo;
	w[fy]=w[fy]*qpow(w[fx],mo-2)%mo;
  }
  else
  {
    int fx=find(u);
    ll ans=w[u]*tot%mo;
	if(u!=fx)
	printf("%lld\n",ans*w[fx]%mo); 
	else printf("%lld\n",ans);
  }
 } 
 return 0;	
}

 

标签:mo,int,ll,father,ans,include,动物园,ZR
来源: https://www.cnblogs.com/lihan123/p/11740824.html