Good Bye 2017 D. New Year and Arbitrary Arrangement

最新推荐文章于 2021-01-19 12:24:15 发布

原创最新推荐文章于 2021-01-19 12:24:15 发布 · 267 阅读

0 ·

CC 4.0 BY-SA版权

动态规划专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一个使用动态规划解决带有概率计算的问题。通过定义状态dp[i][j]为当前有i个a字符和j个ab子序列的情况，推导出了状态转移方程。特别地，当i+j大于等于特定阈值k时，采用了几何分布进行计算。

看了别人的题解
首先这题是一个dp

dp[i][j] i是当前有多少个a j是当前有多少个ab子序列

dp[i][j] = dp[i+1][j]*Pa + dp[i][i+j]*Pb;
i,j 时加一个a之后会变成i+1, j
i,j 时加一个b之后会变成i, i+j

除此之外的话对于i+j >= k的情况
其实是一个几何分布来概括，此时
dp[i][j] = i+j + 1/p - 1

#include<iostream>
#include<map>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<set>
#include<vector>
#include<queue>
#include<cmath>
#include<stack>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 1e3+5;
const int MOD = 1e9+7;
#define MS(x,y) memset(x,y,sizeof(x))
#define MP(x, y) make_pair(x, y)
const int INF = 0x3f3f3f3f;

ll dp[N][N];

ll inv(ll a) {
    ll ans = 1; int tim = MOD-2; 
    while(tim)  {
        if(tim & 1) ans = ans*a % MOD;
        a = a*a % MOD;
        tim >>= 1;
    }
    return ans;
}


int main() {
    int k; ll a, b;
    while(~scanf("%d %lld %lld", &k, &a, &b)) {
        memset(dp, 0, sizeof(dp));
        ll all = a+b;

        a = a*inv(all) %MOD;
        b = b*inv(all) %MOD;
        ll tt = a*inv(b) % MOD;

        for(int i = k; i >= 0; --i) {
            for(int j = k; j >= 0; --j) {
                if(i + j >= k) {
                    dp[i][j] = (i + j + tt) % MOD;
                } else {
                    dp[i][j] = (dp[i + 1][j]*a + dp[i][i + j]*b) % MOD;
                }
            }
        }

        printf("%lld\n", dp[1][0]);
    }
    return 0;
}