BZOJ 4008 HNOI2015 亚瑟王

Problem

BZOJ

Solution

这道题做得真的很懵逼，我至今还记得上个月在做概率与期望的时候，因为这道题而荒废了一个下午。另外还要感谢Chlience大佬给我讲解思路qwqqq

有关状态的定义比较难想到。我们设f[i][j]表示r轮中，前i张卡出了j张卡的概率，显然要满足$i\geq j$，才是有效状态。至于这么设状态的原因，我们会发现“立即结束该轮”的条件很棘手，这导致后面的卡牌可能不会考虑其发动概率。那么我们不妨考虑加一维状态，表示仅考虑前i张卡牌的情况，这会方便我们进行状态转移。

首先我们要明确一点，卡牌i始终打不出来的概率是$pow(1-p[i],r)$，相应的，我们用全集减一下就能得到，卡牌i打出来的概率是$1-pow(1-p[i],r)$。

那么每张牌的打出概率为$ex[i]=\sum_{j=0}^rf[i-1][j]*(1-pow(1-p[i],r-j))$

由上，第一张牌前面没有牌，所以不会受到前面牌是否打出的影响，我们容易得到边界条件：$f[1][0]=pow(1-p[1],r)$，$f[1][1]=1-f[1][0]$。

对于f[i][j]，我们将答案从f[i-1][]转移而来时，就只需要考虑第i张牌究竟能不能打出来。

对于打不出来的情况：$f[i][j]+=f[i-1][j]*pow(1-p[i],r-j)$。因为前i-1张牌打出了j张，那么就有r-j轮会扫到第i张牌。
同理，对于打出来的情况：$f[i][j]+=f[i-1][j-1]*(1-pow(1-p[i],r-j+1))$

但在状态转移时要注意满足f[i-1][j]为有效状态，即$i>j$。

Code

代码跑了6648ms，如果你对每组询问预处理幂而不是每次调用pow的话会快很多，大概1000ms左右。

#include <cstdio>
#include <cmath>
using namespace std;
int z,n,m,inj[250];
double ans,p[250],ex[250],f[250][250];
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif
    scanf("%d",&z);
    while(z--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%lf%d",&p[i],&inj[i]);
            ex[i]=0.0;
        }
        f[1][0]=pow(1-p[1],m);ex[1]=f[1][1]=1.0-f[1][0];
        ans=ex[1]*inj[1];
        for(int i=2;i<=n;i++)
        {
            for(int j=0;j<=i&&j<=m;j++)
            {
                f[i][j]=0.0;
                if(j) f[i][j]+=f[i-1][j-1]*(1.0-pow(1.0-p[i],m-j+1));//抽到了
                if(i>j)
                {
                    f[i][j]+=f[i-1][j]*pow(1.0-p[i],m-j);//抽不到
                    ex[i]+=f[i-1][j]*(1.0-pow(1.0-p[i],m-j));
                }
            }
            ans+=ex[i]*inj[i];
        }
        printf("%.10lf\n",ans);
    }
    return 0;
}