CodeForces 708E Student‘s Camp 题解

题目传送门

题目大意： 有一个 $n+2$ 行 $m$ 列的矩阵，上下两行不能被侵蚀，下面经过 $k$ 天，每天每一行最左边和最右边的一个格子可能会被侵蚀，问最后最上面一行和最下面一行连通的概率是多少？

题解

相当经典的前缀和优化dp。

假如 $k$ 天后，一行的前面被侵蚀了 $i$ 个，这种情况的概率就是 $A_i=\left(\genfrac{}{}{0ex}{}{k}{i}\right)p^i(1-p{)}^{k-i}$，可以预处理出 $A$。

然后先考虑一个暴力dp：$f_{i,l,r}$ 表示从上到下第 $i$ 行，$k$ 天后剩下的区间为 $[l,r]$，并且这个剩下部分与最上面一行连通的概率，转移为 $f_{i,l,r}=A(l-1)A(m-r){\sum }_{[l^{\prime },r^{\prime }]\cap [l,r]\ne \varphi }{f}_{i-1,l^{\prime },r^{\prime }}$。

考虑优化这个dp，注意到下面的限制本质是 $l^{\prime }\le r\&\&r^{\prime }\ge l$，那么就可以将转移表示成另外一种形式：
$$f_{i,l,r}=A_{l-1}{A}_{m-r}\left(\left(\sum _{1\le l^{\prime }\le r^{\prime }\le m}{f}_{i-1,l^{\prime },r^{\prime }}\right)-\left(\sum _{1\le l^{\prime }\le r^{\prime }<l}{f}_{i-1,l^{\prime },r^{\prime }}\right)-\left(\sum _{r<l^{\prime }\le r^{\prime }\le m}{f}_{i-1,l^{\prime },r^{\prime }}\right)\right)$$

不妨记 $pr{e}_{i,p}={\sum }_{1\le l\le r\le p}{f}_{i,l,r},su{f}_{i,p}={\sum }_{p\le l\le r\le m}{f}_{i,l,r}$，代入变成了很简洁的东西：
$$f_{i,l,r}=A_{l-1}{A}_{m-r}(pr{e}_{i-1,m}-pr{e}_{i-1,l-1}-su{f}_{i-1,r+1})$$

考虑跳过 $f$，直接尝试进行递推 $pre$，展开 $pre$ 得到：
$$pr{e}_{i,p}=\sum _{l=1}^p\sum _{r=l}^p{A}_{l-1}{A}_{m-r}(pr{e}_{i-1,m}-pr{e}_{i-1,l-1}-su{f}_{i-1,r+1})$$

不难想到 $pr{e}_{i,p}$ 是可以从 $pr{e}_{i,p-1}$ 递推过来的，即：
$$\begin{array}{rl}pr{e}_{i,p}& =pr{e}_{i,p-1}+\sum _{l=1}^p{A}_{l-1}{A}_{m-p}(pr{e}_{i-1,m}-pr{e}_{i-1,l-1}-su{f}_{i-1,p+1})\\ & =pr{e}_{i,p-1}+\left(A_{m-p}pr{e}_{i-1,m}\sum _{l=1}^p{A}_{l-1}\right)-\left(A_{m-p}\sum _{l=1}^p{A}_{l-1}pr{e}_{i-1,p-1}\right)-\left(A_{m-p}su{f}_{i-1,p+1}\sum _{l=1}^p{A}_{l-1}\right)\end{array}$$

不难发现三个 $\sum$ 都是可以使用前缀和优化的，于是就做完了。

$suf$ 的递推是类似的，可以自己手玩一下，或者直接看代码。

边界为 $pr{e}_{0,m}=su{f}_{0,1}=1$，答案为 $pr{e}_{n,m}$，时间复杂度 $O(nm+k)$。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define mod 1000000007

int n,m,pa,pb,p,k;
int fac[100010],inv_fac[100010],d1[100010],d2[100010],A[2010];
int ksm(int x,int y){int re=1;for(;(y&1?re=1ll*re*x%mod:0),y;y>>=1,x=1ll*x*x%mod);return re;}
void FacInit(){
	fac[0]=inv_fac[0]=1;
	for(int i=1;i<=k;i++)fac[i]=1ll*fac[i-1]*i%mod;
	inv_fac[k]=ksm(fac[k],mod-2);
	for(int i=k-1;i>=1;i--)inv_fac[i]=1ll*inv_fac[i+1]*(i+1)%mod;
}
int Binom(int x,int y){
	if(x<y||y<0)return 0;
	return 1ll*fac[x]*inv_fac[y]%mod*inv_fac[x-y]%mod;
}
void Init(){
	p=1ll*pa*ksm(pb,mod-2)%mod;d1[0]=1;
	for(int i=1;i<=k;i++)d1[i]=1ll*d1[i-1]*p%mod;
	p=(1-p+mod)%mod;d2[0]=1;
	for(int i=1;i<=k;i++)d2[i]=1ll*d2[i-1]*p%mod;
	FacInit();for(int i=0;i<=m&&i<=k;i++)
		A[i]=1ll*Binom(k,i)*d1[i]%mod*d2[k-i]%mod;
}
int pre[2][2010],suf[2][2010],cur=0;
int sumA[2010],Ap[2010],As[2010];
void add(int &x,int y){x=(x+y>=mod?x+y-mod:x+y);}
int add(int x){return x>=mod?x-mod:x;}
void dp(){
	sumA[0]=A[0];for(int i=1;i<=m;i++)
		sumA[i]=add(sumA[i-1]+A[i]);
	pre[cur][m]=suf[cur][1]=1;
	while(n--){
		cur^=1;
		/*memset(pre[cur],0,sizeof(pre[cur]));
		memset(suf[cur],0,sizeof(suf[cur]));
		for(int i=1;i<=m;i++)
			for(int l=1;l<=i;l++)
				for(int r=l;r<=i;r++)
					add(pre[cur][i],1ll*A[l-1]*A[m-r]%mod*(pre[cur^1][m]-add(pre[cur^1][l-1]+suf[cur^1][r+1])+mod)%mod);
		for(int i=1;i<=m;i++)
			for(int l=i;l<=m;l++)
				for(int r=l;r<=m;r++)
					add(suf[cur][i],1ll*A[l-1]*A[m-r]%mod*(pre[cur^1][m]-add(pre[cur^1][l-1]+suf[cur^1][r+1])+mod)%mod);*/
		for(int i=1;i<=m;i++){
			pre[cur][i]=(1ll*pre[cur^1][m]*A[m-i]%mod*sumA[i-1]%mod
						-1ll*A[m-i]*Ap[i]%mod+mod
						-1ll*A[m-i]*suf[cur^1][i+1]%mod*sumA[i-1]%mod+mod)%mod;
			suf[cur][i]=(1ll*A[i-1]*pre[cur^1][m]%mod*sumA[m-i]%mod
						-1ll*A[i-1]*pre[cur^1][i-1]%mod*sumA[m-i]%mod+mod
						-1ll*A[i-1]*As[i]%mod+mod)%mod;
		}
		pre[cur][0]=suf[cur][m+1]=0;
		for(int i=1;i<=m;i++)
			add(pre[cur][i],pre[cur][i-1]);
		for(int i=m;i>=1;i--)
			add(suf[cur][i],suf[cur][i+1]);
		Ap[0]=0;As[m+1]=0;
		for(int i=1;i<=m;i++)
			Ap[i]=add(Ap[i-1]+1ll*A[i-1]*pre[cur][i-1]%mod);
		for(int i=m;i>=1;i--)
			As[i]=add(As[i+1]+1ll*A[m-i]*suf[cur][i+1]%mod);
	}
}

int main()
{
	scanf("%d %d %d %d %d",&n,&m,&pa,&pb,&k);
	Init();dp();printf("%d",pre[cur][m]);
}