CF908D New Year and Arbitrary Arrangement [概率DP]

传送门

f[i][j] 表示有 i 个 a, j 个 ab 的期望, 设 A 为选 a 的概率, B 为选 b 的

如果 i + j >= k 那么再放一个 b 就可以结束, 这种情况的期望就是

转移的话

初始状态如果是 0, 0 的话

显然会死循, 因为一开始可能为 bbbbbbb...

但由于 A = 1 - B, 所以上面那个式子等价于 

所以初始状态设为 f[1][0] 即可

---

#include<bits/stdc++.h>
#define N 1050
using namespace std;
typedef long long ll;
const int Mod = 1000000007;
int k; ll pa, pb, f[N][N];
ll add(ll a, ll b){ return (a+b) % Mod;}
ll mul(ll a, ll b){ return (a*b) % Mod;}
ll power(ll a, ll b){
	ll ans = 1; for(;b;b>>=1){
		if(b&1) ans = mul(ans, a);
		a = mul(a, a);
	} return ans;
}
int main(){
	scanf("%d%lld%lld", &k, &pa, &pb);
	ll A = mul(power(pa + pb, Mod - 2), pa);
	ll B = mul(power(pa + pb, Mod - 2), pb);
	ll C = mul(A, power(Mod + 1 - A, Mod - 2));
	for(int i=k; i>=1; i--){
		for(int j=k; j>=0; j--){
			if(i + j >= k) f[i][j] = add(add(i, j), C);
			else f[i][j] = add(mul(f[i+1][j], A), mul(f[i][i+j], B));
		}
	} printf("%lld", f[1][0]); return 0;
}