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

0|1前言

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

---

1|0正题

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

---

1|1题目大意

$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$

---

1|2解题思路

只考虑外向树的话就是水题了，因为显然的$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)$

---

1|3code

#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;
}

__EOF__

本文作者：QuantAsk
本文链接：https://www.cnblogs.com/QuantAsk/p/14974373.html
关于博主：退役OIer，GD划水选手
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！