CF908D New Year and Arbitrary Arrangement

一、题目

点此看题

你谷的翻译有点小问题，本题求的是$ab$子序列个数的期望

二、解法

设$dp[i][j]$为前缀有$i$个$a$字符，$j$个$ab$子序列，最后子序列个数超过$k$的子序列个数期望，转移如下：
$$dp[i][j]=\frac{dp[i+1][j]\times pa+dp[i][i+j]\times pb}{pa+pb}$$但是你发现转移存在一个很大的问题，就是没有边界条件，我们考虑一种特殊的情况，也就是$i+j\ge k$，此时再加入一个$b$就可以结束，我们尝试把它作为边界条件，现在就来推式子，首先它的期望为：
$$\frac{pb}{pa+pb}\sum _{a=0}^{\infty }(i+j+a)(\frac{pa}{pa+pb}{)}^a$$此时使用错位相减法，我们设上式为$f$，设$x=\frac{pa}{pa+pb}$，我们用$f=\frac{f-xf}{1-x}$来表示：
$$=(1-x)\frac{{\sum }_{a=0}^{\infty }(i+j+a)x^a-{\sum }_{a=0}^{\infty }(i+j+a)x^{a+1}}{1-x}$$$$=\sum _{a=0}^{\infty }(i+j+a)x^a-\sum _{a=0}^{\infty }(i+j+a)x^{a+1}$$$$=i+j+\sum _{a=1}^{\infty }{x}^a$$再次对最后一项使用错位相减法：
$$=i+j+\frac{x}{1-x}=i+j+\frac{pa}{pb}$$那么边界条件就有了，就可以$dp$了，最后还有一个小问题，以$dp[0][0]$开始的话会出现自己转移自己的情况，看一下转移方程就能推出$dp[0][0]=dp[1][0]$，然后我们从$dp[1][0]$开始搜索就行了，时间复杂度$O(k^2)$，贴个代码$qwq$

#include <cstdio>
#define int long long
const int M = 1005;
const int jzm = 1e9+7;
int read()
{
    int x=0,flag=1;
    char c;
    while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
    while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return x*flag;
}
int k,pa,pb,inv1,inv2,dp[M][M];
int qkpow(int a,int b)
{
    int r=1;
    while(b>0)
    {
        if(b&1) r=r*a%jzm;
        a=a*a%jzm;
        b>>=1;
    }
    return r;
}
int dfs(int x,int y)
{
    if(dp[x][y]) return dp[x][y];
    if(x+y>=k) return (x+y+pa*inv1%jzm)%jzm;
    return dp[x][y]=(dfs(x+1,y)*pa+dfs(x,x+y)*pb)%jzm*inv2%jzm;
}
signed main()
{
    k=read();pa=read();pb=read();
    inv1=qkpow(pb,jzm-2);
    inv2=qkpow(pa+pb,jzm-2);
    printf("%lld\n",dfs(1,0));
}