P5405-[CTS2019]氪金手游【树形dp,容斥,数学期望】

前言

话说在\(Loj\)下了个数据发现这题的名字叫\(fgo\)

---

正题

题目链接:https://www.luogu.com.cn/problem/P5405

---

题目大意

\(n\)张卡的权值为\(1/2/3\)的概率权重分别是\(p_{x,1/2/3}\)，然后按照权值每次获得一张未获得的卡，然后再该出一棵有向树（方向可以都是外向或内向的），求所有每条边\((u,v)\)，\(u\)都比\(v\)先获得的概率。

\(1\leq n\leq 1000,0\leq p_{i,j}\leq 10^6\)

---

解题思路

只考虑外向树的话就是水题了，因为显然的\(x\)要排在子树最前面的概率就是\(\frac{w_x}{\sum_{y\in subtree_x}w_y}\)。

然后直接\(n^2\)的\(dp\)就可以力。

但是现在有内向边怎么办，还是考虑转换成只有外向的，也就是去掉一种限制。

去掉一种限制的话容斥是一个不错的办法，考虑的话就是恰好若干条指定边（内向边），我们可以指定至少\(k\)跳内向边不满足条件，这样就组成了一个外向森林，可以很容易处理出答案，而且这样的容斥系数就是\((-1)^k\)。

然后直接\(dp\)就得了，设\(f_{i,j}\)表示到节点\(i\)然后权值和是\(j\)，如果限制一条内向边就直接乘上一个\(-1\)就好了。

额这种树形\(dp\)枚举子树大小可以做到\(n^2\)这个是老生常谈了

时间复杂度\(O(n^2)\)

---

code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=1100,P=998244353;
struct node{
	ll to,next,w;
}a[N<<1];
ll n,tot,ans,ls[N],siz[N],w[N][3],f[N][3*N],g[N*3];
ll power(ll x,ll b){
	ll ans=1;
	while(b){
		if(b&1)ans=ans*x%P;
		x=x*x%P;b>>=1;
	}
	return ans;
}
void addl(ll x,ll y,ll w){
	a[++tot].to=y;
	a[tot].next=ls[x];
	a[tot].w=w;
	ls[x]=tot;return;
}
void dp(ll x,ll fa){
	ll d=power(w[x][0]+w[x][1]+w[x][2],P-2);
	siz[x]=3;
	f[x][1]=w[x][0]*d%P;
	f[x][2]=w[x][1]*d*2ll%P;
	f[x][3]=w[x][2]*d*3ll%P;
	for(ll e=ls[x];e;e=a[e].next){
		ll y=a[e].to;
		if(y==fa)continue;
		dp(y,x);
		if(a[e].w){
			for(ll i=1;i<=siz[x];i++)
				for(ll j=1;j<=siz[y];j++)
					(g[i+j]-=f[x][i]*f[y][j]%P)%=P,(g[i]+=f[x][i]*f[y][j]%P)%=P;
		}
		else{
			for(ll i=1;i<=siz[x];i++)
				for(ll j=1;j<=siz[y];j++)
					(g[i+j]+=f[x][i]*f[y][j]%P)%=P;
		}
		siz[x]+=siz[y];
		for(ll i=1;i<=siz[x];i++)
			f[x][i]=g[i],g[i]=0;
	}
	for(int i=1;i<=siz[x];i++)
		f[x][i]=f[x][i]*power(i,P-2)%P;
	return;
}
signed main()
{
	scanf("%lld",&n);
	for(ll i=1;i<=n;i++)
		scanf("%lld%lld%lld",&w[i][0],&w[i][1],&w[i][2]);
	for(ll i=1;i<n;i++){
		ll x,y;scanf("%lld%lld",&x,&y);
		addl(x,y,0);addl(y,x,1);
	}
	dp(1,0);
	for(ll i=1;i<=siz[1];i++)
		(ans+=f[1][i])%=P;
	printf("%lld\n",(ans+P)%P);
	return 0;
}