[CSP-S模拟测试]:玩具(概率DP)
题目描述
这个故事发生在很久以前,在$IcePrincess\text{_}1968$和$IcePrince\text{_}1968$都还在上幼儿园的时候。
$IcePrince\text{_}1968$最近迷上了一种玩具,这种玩具中有两种零件:圆球和棍子。棍子的两头可以插在两个圆球上的各一个空洞中,从而将两个圆球连接起来。为了保证玩具的娱乐性,任意一个圆球上的空洞个数总是多于玩具套装中的棍子数。你可以认为圆球是没有体积的,所有棍子的长度均为$1$。
$IcePrince\text{_}1968$喜欢这样玩这种玩具:他先摸出玩具袋里的一个圆球放在地上,然后重复下面的操作$n-1$次:每次从袋中取出一个圆球和一根棍子,然后等概率的从地上的圆球中选择一个,将该圆球和选择的圆球用棍子连起来,使得新的圆球在选中圆球的正上方。
$IcePrince\text{_}1968$对自己搭出的艺术品很满意,便决定把这个物品送给$IcePrincess\text{_}1968$作为生日礼物。然而生日礼物是需要包装的,因为默认圆球没有体积,所以$IcePrince\text{_}1968$不用考虑包装盒的长和宽,但是包装盒的高是需要确定的,这里我们假设$IcePrince\text{_}1968$是一个非常节俭的孩子,所以包装盒的高总是等于艺术品的高度。$IcePrince\text{_}1968$想知道自己需要的包装盒的高的期望对质数$p$取模后的值,但他还在上幼儿园,怎么会算呢,于是就请你来帮助他。
输入格式
输入文件名为$toy.in$。
输入数据仅一行,包含两个正整数$n,p$,表示最终的艺术品中圆球的个数和模数$p$。
输出格式
输出文件名为$toy.out$。
输入文件仅一行,一个正整数,表示包装盒的高的期望对质数$p$取模后的值。
样例
样例输入:
3 998244353
样例输出:
499122178
数据范围与提示
样例解释:
三个圆球组成的艺术品,高度只可能是$1$或者$2$,所以高度的期望是$1.5$,在模$998244353$下的期望是$499122178$。
数据范围:
对于$30\%$的数据,满足$n\leqslant 10,p\leqslant 1,000,007$;
对于$50\%$的数据,满足$n\leqslant 20$;
对于$70\%$的数据,满足$n\leqslant 50$;
对于$100\%$的数据,满足$n\leqslant 200,p\leqslant 1,000,000,007$,$p$是质数。
题解
超级神奇的$DP$,先作出如下定义:
$\alpha.$设$dp[i][j]$表示$i$个点的森林,有$j$个点在第一棵树的概率。
$\beta.$设$f[i][j]$表示有$i$个点的树,深度不超过$j$的概率。
$\gamma.$设$g[i][j]$表示有$i$个点的森林,深度不超过$j$的概率。
那么,先考虑如何求出$dp$数组,有:
$$dp[i][j]=dp[i-1][j-1]\times\frac{j-1}{i}+dp[i-1][j]\times\frac{i-j}{i}$$
初始化$dp[1][1]=1$。
解释一下,式子的前一半是将新加入的点放入了第一棵树,那么它可以接到$(j-1)$个点上, 后一半是将新加入的这个点放入了除第一棵树外的其它树上(也可以新建一棵树),分母是$i$而不是$(i-1)$的是由于这个点可以新建一棵树,也就是相当于跟谁也没有相接。
接着考虑一个式子:
$$f[i][j]=g[i-1][j-1]$$
化个图你就理解了$\downarrow$
$$\Downarrow$$
也就是说,我们加了一个点,让这棵森林中的所有树都连到了这个节点上。
再来考虑如何求出$g$数组:
$$g[i][j]=\sum \limits_{k=1}^if[k][i]\times g[i-1][j]\times dp[i][k]$$
这相当于是在$i$个点中选取了$k$个点组成第一棵树,剩下$k$个点还是一个森林,$dp[i][k]$相当于是系数,也就是这第一棵树的形态数。
最后,因为$f$数组的意义是$f[i][j]$有$i$个点的树,深度不超过$j$的概率;所以我们需要知道深度恰好为$j$的概率,即为$f[n][j]-f[n][j-1]$。
将每一个深度的概率乘上其深度再加和即为答案。
时间复杂度:$\Theta(n^3)$。
期望得分:$100$分。
实际得分:$100$分。
代码时刻
#include<bits/stdc++.h>
using namespace std;
int n,p;
long long inv[201];
long long dp[201][201],f[201][201],g[201][201];
long long ans;
long long qpow(long long x,long long y)
{
long long res=1;
while(y)
{
if(y&1)res=res*x%p;
x=x*x%p;
y>>=1;
}
return res;
}
int main()
{
scanf("%d%d",&n,&p);
dp[1][1]=f[1][0]=1;
for(int i=1;i<=n;i++)inv[i]=qpow(i,p-2);
for(int i=2;i<=n;i++)
for(int j=1;j<=i;j++)
dp[i][j]=(dp[i-1][j-1]*(j-1)%p+dp[i-1][j]*(i-j)%p)*inv[i]%p;
for(int i=0;i<=n;i++)g[0][i]=1;
for(int i=1;i<=n;i++)
for(int j=0;j<=n;j++)
{
if(j)f[i][j]=g[i-1][j-1];
for(int k=1;k<=i;k++)g[i][j]=(g[i][j]+f[k][j]*g[i-k][j]%p*dp[i][k]%p)%p;
}
for(int i=1;i<=n;i++)ans=(ans+(f[n][i]-f[n][i-1]+p)%p*i%p)%p;
printf("%lld",ans);
return 0;
}
rp++