守卫者的挑战

https://www.zybuluo.com/ysner/note/1239414

题面

初始值为(k)，现经过一段数，有(p_i)的概率加上第(i)个数(a_i)。问最后值(geq0)且加了(l)个数的概率。

• (0leq kleq2000,0leq nleq200,-1leq a_ileq1000)

解析

注意到(k>200)都是没有意义的，于是可以把(k)的范围视为(0leq kleq200)。
直接设(f[i][j][k])表示处理了第(1-i)个数，加了(j)个数，值为(k)的概率。
决策就是是否加上这个数。

为了防止第三维变负，(k+=200)，空间要开到(400)。
注意到因为(double)有(8)字节，(128MB)下该数组会导致(MLE)。
滚掉第一维即可。

复杂度(O(n^3))（(n,k)同阶）。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#define re register
#define il inline
#define ll long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define fp(i,a,b) for(re int i=a;i<=b;i++)
#define fq(i,a,b) for(re int i=a;i>=b;i--)
using namespace std;
const int mod=1e9+7,N=1e6+100;
int n,l,k,sum;
long double dp[2][205][410],ans;
struct cha{int w;long double p;}a[N];
il ll gi()
{
   re ll x=0,t=1;
   re char ch=getchar();
   while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
   if(ch=='-') t=-1,ch=getchar();
   while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
   return x*t;
}
il int del(re int x)
{
  if(x>400) return 400;
}
int main()
{
  freopen("guard.in","r",stdin);
  freopen("guard.out","w",stdout);
  n=gi();l=gi();k=gi();k=min(k+200,400);
  fp(i,1,n) scanf("%Lf",&a[i].p);fp(i,1,n) a[i].w=gi();
  re int now=1,nxt=0;
  fp(i,200,400) fp(j,l,n) dp[now][j][i]=1;
  fq(i,n,1)
  {
    swap(now,nxt);
    memset(dp[now],0,sizeof(dp[now]));
    fp(j,0,i)
      fp(v,0,400)
      {
	dp[now][j][v]+=a[i].p*dp[nxt][j+1][del(v+a[i].w)]/100;
	dp[now][j][v]+=(100-a[i].p)*dp[nxt][j][v]/100;
      }
  }
  printf("%.6Lf
",dp[now][0][k]);
  fclose(stdin);
  fclose(stdout);
  return 0;
}