毕业生的序列游戏

题目：
https://codeforces.ml/problemset/problem/908/D

有一个字符串，初始为空，每次随机添加$a$或$b$，当子序列为$ab$的个数大于等于$k$时停止。求最终得到的子序列$ab$的个数的期望。

思路：
设$f(i,j)$表示从字符串有$i$个$a$，$j$个$ab$时的状态，到最终态(即$ab$的个数大于等于$k$)的期望个数(期望$dp$的套路，反着$dp$)，有$p_a$概率添加a，同理$p_b$，$p_a+p_b=1$则有
$$f(i,j)=p_a\ast f(i+1,j)+p_b\ast (i,j+i)$$
解释就是$i$个$a$，$j$个$ab$时的状态，添加$a$就会到达$i+1$个$a$，$j$个$ab$时的状态,添加$b$就会到达$i$个$a$，$i+j$个$ab$时的状态。答案就是$f(0,0)$。
现在考虑初始值

• 当$j\ge k$时
  $$f(i,j)=j$$
  因为已经结束了，得到的$ab$一定是$j$个。但这样还不够，比如$f(0,0)$和$f(1,0)$有关，$f(1,0)$和$f(2,0)$有关，一直到$f(\infty ,0)$。
• 当$i+j\ge k,j<k$时，再添加$x(0\le x\le \infty )$个连续的$a$，然后在添加一个$b$就能结束状态，最后得到$i+j+x$个$ab$，所以有
  $$\begin{array}{rl}f(i,j)& =\sum _{x=0}^{\infty }{p}_a^x{p}_b(i+j+x)\\ & =(i+j)p_b\sum _{x=0}^{\infty }{p}_a^x+p_b\sum _{x=0}^{\infty }x{p}_a^x(p_a^{\infty }=0)\\ & =(i+j)p_b\frac{1}{1-pa}+p_b\sum _{x=0}^{\infty }x{p}_a^x\end{array}$$
  $$\begin{array}{rl}\sum _{x=0}^{\infty }x{p}_a^x& =1p_a^1+2p_a^2+3p_a^3+...+\infty p_a^{\infty }\\ & =(p_a^1+p_a^2+p_a^3+...p_a^{\infty })+(p_a^2+p_a^3+...p_a^{\infty })+(p_a^3+...p_a^{\infty })\\ & =\sum _{i=1}^{\infty }\sum _{j=i}^{\infty }{p}_a^j\\ & =\frac{p_a}{(1-p_a{)}^2}\end{array}$$
  所以
  $$f(i,j)=i+j+\frac{p_a}{p_b}$$
  所以要从$f(0,0)$递归$dp$计算，因为初始值有无穷多，没法从初始状态推。

#include<cstdio>
#include<cstring>
const int MAXK=1005,MOD=1000000007;

int pow_mod(int a,int b)
{
    int res=1;
    while(b)
    {
        if(b&1)
            res=(1LL*res*a)%MOD;
        b>>=1;
        a=(1LL*a*a)%MOD;
    }
    return res;
}
int inv(int x)
{return pow_mod(x,MOD-2);}

int dp[MAXK][MAXK];
int k,pa,pb;

int DP(int i,int j)
{
    if(j>=k)
        return j;
    if(i+j>=k)
        return (i+j+1LL*pa*inv(pb)%MOD)%MOD;
    if(dp[i][j]!=-1)
        return dp[i][j];
    return dp[i][j]=(1LL*((1LL*pa*DP(i+1,j))%MOD+(1LL*pb*DP(i,i+j))%MOD)*inv(pa+pb))%MOD;
}

int main()
{
    scanf("%d%d%d",&k,&pa,&pb);
    memset(dp,-1,sizeof dp);
    printf("%d\n",DP(1,0));
    return 0;
}