NOIP模拟赛20190914 T3 空之轨迹【概率DP】

题目描述：

简化版题意：
随机生成一个 $m+1$ 个数的数列，第一个数为 $0$, 生成第 $i$ 个数时，在前 $i-1$ 个数中等概率选择一个数 $k$, 则第 $i$ 个数为 $k+1$。数字 $i$ 有一个对应的权值 $a_i$，求数列权值和的期望。
$m\le 21$，$\mathrm{m}\mathrm{o}\mathrm{d}998244353$

题目分析：

（我所知的）大概有两种方法。

法一：排序差分

如果直接算期望转移的话，就要想办法表达已经选的前$i$个数是那些数。

显然数的顺序是不重要的，我们只需要知道每种数有多少。将所得的$i$个数拿来排序后，可以发现相邻两个数的差只能是0或1，所以就可以用一个$2^m$的状态来表示已经有的数。
设$f[i]$表示已有数字状态为$i$，还能再得到的权值和的期望，答案就是$f[0]$。
可以发现我们的转移每次都是涉及到状态中的所有数，所以不必将当前拿的哪一个数放进状态中。
只需枚举当前选的数字$j$，记$i$中数的个数为$cn{t}_i$，则$f[i]+=\frac{f[iaddj]}{cn{t}_i}$
复杂度$O(m\cdot 2^m)$

Code：

#include<cstdio>
#define maxn 25
using namespace std;
const int mod = 998244353;
int n,m,a[maxn],f[1<<21],lg[1<<21],inv[maxn];
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++) scanf("%d",&a[i]);
	inv[0]=inv[1]=1;
	for(int i=2;i<=m;i++) inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
	for(int i=2;i<1<<(m-1);i++) lg[i]=lg[i>>1]+1;
	for(int s=(1<<(m-1))-1;s>=1;s--){//begin shade 01, end have a 1 as barrier.
		f[s]=f[s<<1]+a[0];//origin 0->1
		for(int t=s,now,pre=-1,low,j=1;t;pre=now,j++){
			now=lg[low=t&-t], t^=low;
			if(t) f[s]=(f[s]+1ll*(f[t<<1|(s^t)]+a[j])*(now-pre))%mod;
			else f[s]=(f[s]+1ll*(f[low<<1|s]+a[j])*(now-pre))%mod;
		}
		f[s]=1ll*f[s]*inv[lg[s]+2]%mod;
	}
	printf("%d\n",(f[1]+a[0])%mod);
}

---

法二：期望的线性性（分开算每个点的概率*权值再累加）

最后的序列是一个长度为$m+1$的序列，设其权值和为$S$
则$S={\sum }_{i=1}^{m+1}{X}_i$，$X_i$为第$i$个位置的权值
那么$E(S)={\sum }_{i=1}^{m+1}E(X_i)$
而$E(X_i)={\sum }_{j=1}^{i-1}p[i][j]\ast a[j]$，$p[i][j]$为第$i$个数为$j$的概率。

于是只需要求出$p[i][j]$。要算这个，我们就需要知道前$i-1$个数中选了多少个$j-1$，对应就会生成多少个$j$，$p[i][j]=\frac{E(生成的j的个数)}{i-1}$，同样分开统计每个位置的贡献，分子的$E={\sum }_{k=1}^{i-1}p[k][j-1]$。

所以就是$p[i][j]=\frac{1}{i-1}\ast {\sum }_{k=1}^{i-1}p[k][j-1]$，用前缀和优化一下就可以做到$O(m^2)$

Code：

#include<cstdio>
#define maxn 25
using namespace std;
const int mod = 998244353;
int n,m,a[maxn],p[maxn][maxn],inv[maxn],ans,sum[maxn];
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++) scanf("%d",&a[i]);
	inv[0]=inv[1]=1;
	for(int i=2;i<=m;i++) inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
	p[1][0]=sum[0]=1;
	for(int i=2;i<=m+1;i++)
		for(int j=i-1;j>=1;j--){
			p[i][j]=1ll*sum[j-1]*inv[i-1]%mod,sum[j]=(sum[j]+p[i][j])%mod;
			ans=(ans+1ll*p[i][j]*a[j])%mod;
		}
	printf("%d\n",ans);
}