1. 程式人生 > >bzoj 3029 守衛者的挑戰——概率期望dp+狀態數思考

bzoj 3029 守衛者的挑戰——概率期望dp+狀態數思考

www. 發現 targe get 有關 lan clu else print

題目:https://www.lydsy.com/JudgeOnline/problem.php?id=3029

先隨便寫了個dfs,記錄“前 i 次、成功 j 次、容量-殘片=k”的概率。因為是否可行只和“成功次數”還有“容量-殘片個數”有關,和容量、殘片具體數量無關。準備記憶化,但發現狀態存不下。

技術分享圖片
#include<iostream>
#include<cstdio>
#include<cstring>
#define db double
using namespace std;
const int N=205,M=202005;
int n,l,K,a[N];
db p[N],ans;
void dfs(int i,int j,int c,double w)//至少有c個 { if(c>n)c=n; if(i>n) { if(j>=l&&c>=0)ans+=w;return; } if(j+(n-i+1)<l)return; if(a[i]>=0) { dfs(i+1,j+1,c+a[i],w*p[i]); dfs(i+1,j,c,w*(1-p[i])); } else { dfs(i+1
,j+1,c-1,w*p[i]); dfs(i+1,j,c,w*(1-p[i])); } } int main() { scanf("%d%d%d",&n,&l,&K);int x; for(int i=1;i<=n;i++) { scanf("%d",&x);p[i]=(db)x/100; } for(int i=1;i<=n;i++)scanf("%d",&a[i]); dfs(1,0,K,1); printf("%.6lf\n",ans);
return 0; }
View Code

然後想到了 bzoj4247 掛飾 的思路。就是發現“容量-殘片數量”大於n的話多出來的部分沒什麽用,可以把大於n記成等於n。這樣第三維就只有2*n啦!然後就可以刷表了。

#include<iostream>
#include<cstdio>
#include<cstring>
#define db double
using namespace std;
const int N=205,M=202005;
int n,l,K,a[N];
db p[N],ans,dp[2][N][N<<1];
int main()
{
    scanf("%d%d%d",&n,&l,&K);int x;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x);p[i]=(db)x/100;
    }
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    dp[0][0][min(n,K)+N]=1;
    for(int i=0;i<n;i++)
    {
        int d=(i&1);
        for(int j=0;j<i;j++)
            for(int k=-n;k<=n;k++)dp[!d][j][k+N]=0;
        for(int j=0;j<=i;j++)
            for(int k=-n;k<=n;k++)
            {
                if(a[i+1]>=0)
                {
                    dp[!d][j+1][min(n,k+a[i+1])+N]+=dp[d][j][k+N]*p[i+1];//+=
                    dp[!d][j][k+N]+=dp[d][j][k+N]*(1-p[i+1]);
                }
                else
                {
                    dp[!d][j+1][k-1+N]+=dp[d][j][k+N]*p[i+1];
                    dp[!d][j][k+N]+=dp[d][j][k+N]*(1-p[i+1]);
                }
            }
    }
    int d=(n&1);
    for(int j=l;j<=n;j++)
        for(int k=0;k<=n;k++)
            ans+=dp[d][j][k+N];
    printf("%.6lf\n",ans);
    return 0;
}

bzoj 3029 守衛者的挑戰——概率期望dp+狀態數思考