AcWing 232 概率 + DP

题意

传送门 AcWing 232 守卫者的挑战

题解

$dp[i+1][j][k]$ 代表在 $0\cdots i$ 这 $i+1$ 个挑战中，成功了 $j$ 次，背包剩余容量为 $k$ 的概率。显然，状态向前递推比较方便。根据挑战是否成功，得到
$$\{\begin{array}{l}dp[i+1][j+1][k+a[i]]+=dp[i][j][k]\times p[i]\\ dp[i+1][j][k]+=dp[i][j][k]\times (1-p[i])\end{array}$$
时空复杂度 $O(N^2\sum a[i])$，由于地图碎片贡献为 $-1$，至多贡献 $-N$，那么背包剩余容量大于等于 $N$ 即能保证可以带上地图碎片离开擂台，$DP$ 第三维规模缩小为 $O(N)$即可。此时，时空复杂度降为 $O(N^3)$。

需要注意的是，能够带地图碎片离开擂台，只需要保证完成 $N$ 项挑战后背包容量大于地图碎片总容量。那么递推过过程中需要考虑背包容量小于零的情况，为方便起见，将第三维值域修改为 $[-N,N]\Rightarrow [0,2N]$。答案为
$$\sum _{L\le j\le N,N\le k\le 2N}dp[N][j][k]$$

#include <bits/stdc++.h>
using namespace std;
const int maxn = 205;
int N, L, K, A[maxn];
double P[maxn], dp[maxn][maxn][2 * maxn];

int main()
{
    scanf("%d%d%d", &N, &L, &K);
    for (int i = 0; i < N; ++i)
        scanf("%lf", P + i), P[i] /= 100;
    for (int i = 0; i < N; ++i)
        scanf("%d", A + i);
    dp[0][0][N + K] = 1;
    for (int i = 0; i < N; ++i)
        for (int j = 0; j <= i; ++j)
            for (int k = N - j; k <= 2 * N; ++k)
            {
                dp[i + 1][j + 1][min(k + A[i], 2 * N)] += dp[i][j][k] * P[i];
                dp[i + 1][j][k] += dp[i][j][k] * (1 - P[i]);
            }
    double res = 0;
    for (int j = L; j <= N; ++j)
        for (int k = N; k <= 2 * N; ++k)
            res += dp[N][j][k];
    printf("%.6f\n", res);
    return 0;
}