bzoj 3029: 守卫者的挑战概率dp

Posted 2021-01-05 lokiii

以后写dp还是向后转移吧……写的把前面加起来的版本怎么也调不过去
首先注意，因为地图碎片只占1体积，所以>n,<-n的体积是没用的，所以就可以把体积降到n级别，然后用这场胜负像后转移即可，滚动数组优化

#include<iostream>
#include<cstdio>
using namespace std;
const int N=405;
int n,l,m,a[N];
double p[N],f[2][N][N],ans;
int main()
{
    scanf("%d%d%d",&n,&l,&m);
    for(int i=1;i<=n;i++)
        scanf("%lf",&p[i]),p[i]/=100;
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    f[0][0][min(m,n)+n]=1;
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=i;j++)
            for(int k=-n;k<=n;k++)
                f[(i+1)&1][j][k+n]=0;
        for(int j=0;j<=i;j++)
            for(int k=-n;k<=n;k++)
                f[(i+1)&1][j][k+n]+=f[i&1][j][k+n]*(1.0-p[i]),f[(i+1)&1][j+1][min(k+a[i],n)+n]+=f[i&1][j][k+n]*p[i];
    }
    for(int j=l;j<=n;j++)
        for(int k=0;k<=n;k++)
            ans+=f[(n+1)&1][j][k+n];
    printf("%.6f
",ans);
    return 0;
}