[JZOJ6221] 担心 【区间DP】【概率】

Description

有n个人，每个人有水平值$A_i$

每一轮随机挑选剩余的相邻两人单挑，假设两人能力值为$a,b$，则能力值为$a$的人有$\frac{a}{a+b}$的概率获胜，能力值为$b$的人有$\frac{b}{a+b}$的概率获胜。

给出n,m以及A，问第m个人获胜的概率，答案对998244353取模。

$n\le 100000,A\le 10^9$

Solution

一个朴素的$O(n^5)$的DP是这样的

记$f_{i,j,k}$表示当前考虑区间$[i,j]$，获胜的人为$k$

考虑有左右两半两个区间转移而来

假定k在左边的区间，在右边的区间情况类似
$$f_{i,j,k}=\frac{1}{j-i}\sum _{x=k}^{j-1}{f}_{i,x,k}\sum _{p=x}^j{f}_{x,j,p}\frac{a_k}{a_k+a_p}$$
乘上$\frac{1}{j-i}$的原因是只需要保证k和p单挑的这一局是在这个区间内所有比赛的最后一场。

答案显然是$f_{1,n,m}$

考虑优化这个DP，记$f_{i,j}$为考虑区间$[i,j]$，最终$i$获胜的概率，$g_{i,j}$为最终$j$获胜的概率

由于一个位置的左右边的比赛是独立的，因此$[l,r]$内$k$获胜的概率就是$f_{l,k}\times g_{k,r}$

容易得到转移
$$f_{i,j}=\frac{1}{j-i}\sum _{k=i}^{j-1}{f}_{i,k}\sum _{u=k+1}^j{g}_{k,u}{f}_{u,j}\frac{a_i}{a_i+a_u}$$
交换主体，我们发现${\sum }_k{f}_{i,k}{g}_{k,u}\frac{a_i}{a_i+a_u}$这一部分只与i,u有关，可以在枚举j以外预处理。

这样转移就不需要枚举k，复杂度降低到$O(n^3)$，能够通过本题。

事实上还有另外一种做法。

我们考虑将整个过程倒过来，从序列中淘汰变成向序列中塞数。

记$f_{i,j}$表示区间$[i,j]$中只塞进了$i,j$，中间的所有均为空的概率。

最后的答案就是$f_{0,m}\ast f_{m,n+1}$

考虑转移，枚举塞哪一个数进去
$$f_{i,j}=\frac{1}{j-i-1}\sum _{k=i+1}^{j-1}{f}_{i,k}{f}_{k,j}(\frac{a_i}{a_i+a_k}+\frac{a_j}{a_j+a_k})$$
它有可能是被区间的两个端点中的任意一个淘汰，一开始乘上$j-i-1$是由于区间内先塞哪个数都是一样的。

特殊的，我们记$a_0=a_{n+1}=0$

边界条件为$f_{i,i+1}=1$

这样也在$O(n^3)$的时间复杂度内解决了问题，后一种方法理解难度较大，但代码难度相当低。

Code

#include <bits/stdc++.h>
#define fo(i,a,b) for(int i=a;i<=b;++i)
#define fod(i,a,b) for(int i=a;i>=b;--i)
const int N=505;
const int mo=998244353;
typedef long long LL;
using namespace std;
int n,m,a[N];
LL f[N][N],ns[N][N],ny[N];
LL ksm(LL k,LL n)
{
	LL s=1;
	for(;n;n>>=1,k=k*k%mo) if(n&1) s=s*k%mo;
	return s;
}
#define inc(x,y) (x=(x+y)%mo)
int main()
{
	cin>>n>>m;
	fo(i,1,n) scanf("%d",&a[i]);
	fo(i,0,n) f[i][i+1]=1;
	fo(i,1,n)
		fo(j,i+1,n) ns[i][j]=ns[j][i]=ksm(a[i]+a[j],mo-2);
	fo(i,1,n) ny[i]=ksm(i,mo-2);
	fo(l,2,n)
	{
		fo(x,0,n-l+1)
		{
			int y=x+l;
			fo(p,x+1,y-1)
			{
				inc(f[x][y],f[x][p]*f[p][y]%mo*((ns[x][p]*a[x]+ns[p][y]*a[y])%mo)%mo);
			} 			
			f[x][y]=f[x][y]*ny[l-1]%mo;
		}
	}
	printf("%lld\n",f[0][m]*f[m][n+1]%mo);
}