Good Bye 2017 D New Year and Arbitrary Arrangement(期望dp+错位相减)

D New Year and Arbitrary Arrangement

题意：

输入$k(1e3),pa,pb$;
初始是空串，每次有$pa/(pa+pb)$的概率在串尾加$a$,$pb/(pa+pb)$的概率在串尾加$b$，当串中子串(位置不一定相邻)为$ab$的个数大于等于$k$时，结束。问结束是$ab$个数期望是多少。

题解：

设$dp[i][j]$为当前$a$的个数为$i$,$ab$的个数为$j$,到结束时$ab$个数的期望。
转移：
$dp[i][j]=(pa\ast dp[i+1][j]+pb\ast dp[i][i+j])/(pa+pb)$
当$i+j>=k$时，通过错位相减可以算出：
$dp[i][j]=i+j+pa/pb$；
最后$dp[1][0]$即为所求。

代码：

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
int k,pa,pb;
int dp[1009][1009];
int poww(int a,int b,int c){
    int ans=1,base=a;
    while(b){
        if(b&1)ans=1ll*ans*base%c;
        base=1ll*base*base%c;
        b>>=1;
    }
    return ans;
}
int main(){
  //  freopen("tt.in","r",stdin),freopen("tt.out","w",stdout);
    cin>>k>>pa>>pb;
    int inva=poww(pa,mod-2,mod),invb=poww(pb,mod-2,mod),invc=poww(pa+pb,mod-2,mod);
    for(int i=k;i>=1;i--)
      for(int j=k;j>=0;j--)
      if(i+j>=k)dp[i][j]=(i+j+1ll*pa*invb)%mod;
      else dp[i][j]=(1ll*pa*invc%mod*dp[i+1][j]+1ll*pb*invc%mod*dp[i][i+j])%mod;
    cout<<dp[1][0]<<endl;
    return 0;
}