codeforces 167B 概率DP

codeforces 167B

---

题意：

$给定一个初始容量为k的背包，你背上它去打n场比赛，第i场比赛获胜的概率为p_i，奖励为a_i。$

• $若a_i=-1，你可以获得体积为1的奖品$
• $若a_i̸=-1，你可以将背包容量扩充a_i$

$问至少获胜l场比赛并且将所有已得奖品带走的概率。$

---

题解：

$dp[i][j][k]表示前i场比赛获胜j场，剩余背包容量为k的概率。$
$为避免状态转移时背包容量出现负值的情况，以200+k代替k。$

• $第i场比赛输，dp[i][j-1][k]=dp[i-1][j-1][k]\ast (1-p[i])$
• $第i场比赛赢且k+a[i]\ge 0时，dp[i][j][k+a[i]]=dp[i-1][j-1][k]\ast p[i]$

---

#include <bits\stdc++.h>
using namespace std;
const int N = 201;
int a[N];
double p[N];
double dp[N][N][N*2];

int main() {
    int n, l, k;
    cin >> n >> l >> k;
    for(int i = 1 ; i <= n ; i++){
        cin >> p[i];
        p[i] /= 100;
    }
    for(int i = 1 ; i <= n ; i++){
        cin >> a[i];
    }
    dp[0][0][200+k] = 1;
    for(int i = 1 ; i <= n ; i++){
        for(int j = 1 ; j <= i ; j++){
            for(int k = 0 ; k <= 400 ; k++){
                dp[i][j-1][k] += dp[i-1][j-1][k]*(1-p[i]);
                int r = min(400, k+a[i]);
                if(r >= 0){
                    dp[i][j][r] += dp[i-1][j-1][k]*p[i];
                }
            }
        }
    }
    double ans = 0;
    for(int j = l ; j <= n ; j++){
        for(int k = 200 ; k <= 400 ; k++){
            ans += dp[n][j][k];
        }
    }
    cout << setiosflags(ios::fixed) << setprecision(12) << ans << endl;
    return 0;
}