CCPC-Wannafly Winter Camp Day3 (Div2, onsite) I 石头剪刀布(按秩合并并查集)

石头剪刀布

题解：每次有两个事件：

1. $y$去挑战$x$，如果赢了可以坐在$x$的位置，打平或者输了就要被淘汰。
2. 询问在进行所有一类事件后，有多少种情况可以让$x$现在还没有被淘汰。

对于第二类事件，我们假设$x$挑战了别人$a$次，被挑战了$b$次，那他没有被淘汰的概率就是$3^n\cdot (\frac{1}{3}{)}^a\cdot (\frac{2}{3}{)}^b$。因此我们只要知道$a,b$的值就可以知道答案。因此我们需要维护$a,b$的值。对于第一类事件，我们画画图就会发现它会形成树状关系。那么我们就可以维护这颗树的节点信息，然后去询问就好了。很容易想到并查集了，但是这里似乎并不能直接用路径压缩，原因听一位巨巨说因为每个节点都有权值，根节点也有权值，每次查询的时候会重复累加根节点的信息，所以需要按秩合并。我们可以维护两个信息：每个人的总比赛场次，主场比赛场次。对于$y$挑战$x$，那么肯定$y,x$的总比赛场次都加$1$，$x$主场比赛场次加$1$。那么，对于集合$y$合并到集合$x$，会将$x$设为$y$的父亲，因为每个节点都包含着它当前这个集合的增量信息，也就意味着集合$y$里的增量会凭空多增加$x$的增量。因此合并的时候需要减去。然后$find$的时候累加增量即可。

补完的心情：说实话这个并查集节点打标记的做法是真的骚啊…合并集合的时候为什么要w[y] -= w[x]理解了好长时间T_T，还有按秩合并也是第一次知道！

代码

#include<bits/stdc++.h>
typedef long long LL;
using namespace std;
const int N = 2E5+10, mod = 998244353;
LL w[N],v[N],rk[N],f[N];
struct node{
	int fa;
	LL _w,_v;
};
LL _pow(LL a,LL n)
{
	LL ret = 1;
	while(n) {
		if(n & 1) ret = a * ret % mod;
		a = a * a % mod;
		n >>= 1;
	}
	return ret;
}

LL Inv(LL n){ return _pow(n, mod - 2); }

node _find(int x)
{
	if(x == f[x]) return node{x,w[x],v[x]};
	node t = _find(f[x]);
	return node{t.fa, t._w + w[x], t._v + v[x]};
}

void _merge(int x,int y)
{
	int t1 = _find(x).fa;
	int t2 = _find(y).fa;
	if(t1 != t2) {
		w[t1]++;
		v[t1]++;
		w[t2]++;
		if(rk[t1] >= rk[t2]) {
			w[t2] -= w[t1];
			v[t2] -= v[t1];
			f[t2] = t1;
			rk[t1]++;
		}else{
			w[t1] -= w[t2];
			v[t1] -= v[t2];
			f[t1] = t2;
			rk[t2]++;
		}
	}
}

int main()
{
#ifndef ONLINE_JUDGE
    freopen("input.in","r",stdin);
#endif
	for(int i = 0; i < N; ++i) {
		f[i] = i;
	}
	int n,m,c,x,y;
	cin >> n >> m;
	LL t = _pow(3,n);
	for(int i = 0; i < m; ++i) {
		scanf("%d",&c);
		if(c == 1) {
			scanf("%d %d",&x,&y);
			_merge(x,y);
		}else{
			scanf("%d",&x);
			node t1 = _find(x);
			LL a = t1._v, b = t1._w - t1._v;
			LL ans = t * Inv(_pow(3,b)) % mod * _pow(2,a) % mod * Inv(_pow(3,a)) % mod;
			printf("%lld\n",ans);
		}
	}
    return 0;
}