CF908D New Year and Arbitrary Arrangement 题解

题目传送门

题目大意： 给出 $k,pa,pb$，一开始有一个空序列，每次有 $\frac{pa}{pa+pb}$ 的概率在末尾放一个 $a$，有 $\frac{pb}{pa+pb}$ 的概率在末尾放一个 $b$，当存在 $k$ 个形如 $ab$ 的子序列时停止，问最后期望有多少个 $ab$ 子序列。

题解

值得注意的是，同一个 $a$ 可以存在于多个子序列 $ab$ 中，$b$ 也是。

那么每一次放 $b$ 时，就要考虑到前面有多少个 $a$，假如有 $j$ 个，那么子序列数量就要加 $j$。那么就能顺理成章的得到 $dp$ 数组的定义：设 $f[i][j]$ 表示有 $i$ 个子序列，$j$ 个 $a$ 的序列的期望答案，方程为：
$$f[i][j]=\frac{pa}{pa+pb}f[i][j+1]+\frac{pb}{pa+pb}f[i+j][j]$$

最后，还要考虑到你是非酋的情况，有可能最后放了很多个 $a$ 都放不出一个 $b$，导致 $j$ 太大然后爆炸，于是我们需要判断，当 $i+j\ge k$ 时，此时但凡只要来一个 $b$ 就能停止。

设 $x=\frac{pa}{pa+pb},y=\frac{pb}{pa+pb}$于是有：
$$\begin{array}{rl}f[i][j]& =i+j+0y+1xy+2x^{2}y+3x^{3}y+...\\ & =i+j+xy(1+2x+3x^2+4x^3+...)\end{array}$$

里面这个柿子是个经典的数列求和，学过生成函数的肯定很眼熟，推导一下就是：
$$1+2x+3x^2+4x^3+...=(1+x+x^2+x^3+...{)}^2$$

右边是一个等比数列，那就好办了，设最后一项为 $n$，则有 $1+x+x^2+x^3+...=\frac{1-x^n}{1-x}$

由于 $0<x<1$，$n$ 趋于无穷大，所以 $x^n$ 趋于 $0$，不妨就当成 $0$，就得到 $1+x+x^2+x^3+...=\frac{1}{1-x}$，于是那个数列就等于 $\frac{1}{(1-x{)}^2}$。

因为 $x+y=1$，所以 $\frac{1}{(1-x{)}^2}=\frac{1}{y^2}$。

带回原柿子得到：
$$f[i][j]=i+j+xy\times \frac{1}{y^2}=i+j+\frac{pa}{pb}$$

于是就可以 $dp$ 了，以及，$dp$ 的时候从 $f[0][1]$ 开始，避免一直出 $b$ 而毫无进展的情况。

代码如下：

#include <cstdio>
#include <cstring>
#define mod 1000000007

int k,a,b,inv_a,inv_b,inv_ab;
int ksm(int x,int y)
{
	int re=1;
	while(y)
	{
		if(y&1)re=1ll*re*x%mod;
		x=1ll*x*x%mod;y>>=1;
	}
	return re;
}
#define inv(x) ksm(x,mod-2)
int f[1010][1010];
int dp(int i,int j)
{
	if(i+j>=k)return (i+j+1ll*a*inv_b%mod)%mod;
	if(f[i][j]!=-1)return f[i][j];
	return f[i][j]=(1ll*a*inv_ab%mod*dp(i,j+1)+1ll*b*inv_ab%mod*dp(i+j,j))%mod;
}

int main()
{
	scanf("%d %d %d",&k,&a,&b);
	inv_a=inv(a);inv_b=inv(b);inv_ab=inv(a+b);
	memset(f,-1,sizeof(f));
	printf("%d",dp(0,1));
}