[BZOJ4008][HNOI2015]亚瑟王期望DP

最新推荐文章于 2021-10-07 14:03:08 发布

原创最新推荐文章于 2021-10-07 14:03:08 发布 · 538 阅读

1 ·

CC 4.0 BY-SA版权

dp 同时被 2 个专栏收录

71 篇文章

订阅专栏

概率与期望

7 篇文章

订阅专栏

记住，期望要倒着推。。。
考虑按一轮一轮DP，不免要记录之前轮发动了那些卡牌，这样就gg了。
所以考虑每一张卡牌的贡献，我们就只需要关心还剩多少轮没有发动卡牌，设 $f_{i,j}$ 表示当前考虑第 $i$ 张卡牌（ $i+1$ 之后的都考虑完了），还剩 $j$ 轮没有发动过，之后能获得的期望伤害。显然对于卡牌 $i$ ，有 $(1-P_i)^j$ 的概率不被发动， $1-(1-P_i)^j$ 的概率被发动，于是转移就是：

f i, j = f i + 1, j * (1 - P i) j + (f i + 1, j - 1 + D i) * (1 - (1 - P i) j)

$f_{i,j}=f_{i+1,j}*(1-P_i)^j+(f_{i+1,j-1}+D_i)*(1-(1-P_i)^j)$
代码：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,r;
double p[230],d[230],f[230][140];
int main()
{
    int ca;
    scanf("%d",&ca);
    while(ca--)
    {
        scanf("%d%d",&n,&r);
        for(int i=1;i<=n;i++)
            scanf("%lf%lf",&p[i],&d[i]);
        memset(f,0,sizeof(f));
        for(int i=n;i;i--)
        {
            double P=1-p[i];
            for(int j=1;j<=r;P*=(1-p[i]),j++)    
                f[i][j]=f[i+1][j]*P+(f[i+1][j-1]+d[i])*(1-P);
        }   
        printf("%.10lf\n",f[1][r]);
    }
    return 0;
}