首页 > 代码库 > BZOJ4374 : Little Elephant and Boxes

BZOJ4374 : Little Elephant and Boxes

设$f[i][j][k]$表示前$i$个物品买了$j$个,消耗$k$个钻石,最少花多少钱,可以通过简单的DP求出。

枚举拥有的钻石数以及最多能购买的物品数的下界,那么钱数的下界是定值。

将$n$个箱子折半搜索,按钻石数分组并排序,枚举左半边每种方案,在右半边双指针求出总概率即可。

时间复杂度$O(nm2^{\frac{n}{2}}+nm^2)$。

 

#include<cstdio>#include<algorithm>const int N=33,inf=~0U>>2;int T,n,m,lim,i,j,k,f[N][N][N],ca[N],cb[N];double p[N],ans;struct P{int x;double p;P(){}P(int _x,double _p){x=_x,p=_p;}}a[N],A[N][33000],B[N][33000];struct E{int c,d;}b[N];inline bool cmp(const P&a,const P&b){return a.x<b.x;}inline void up(int&a,int b){if(a>b)a=b;}void dfsl(int x,int y,int z,double p){  if(x==lim){    A[y][++ca[y]]=P(z,p);    return;  }  dfsl(x+1,y,z+a[x].x,p*a[x].p);  dfsl(x+1,y+1,z,p*(1.0-a[x].p));}void dfsr(int x,int y,int z,double p){  if(x==n){    B[y][++cb[y]]=P(z,p);    return;  }  dfsr(x+1,y,z+a[x].x,p*a[x].p);  dfsr(x+1,y+1,z,p*(1.0-a[x].p));}inline double cal(int x,int y,int z){  if(z>=inf)return 0;  int n=ca[x],m=cb[y],i;double p=0,ret=0;  if(!n||!m)return 0;  for(i=1;i<=n;i++){    while(m&&A[x][i].x+B[y][m].x>=z)p+=B[y][m--].p;    ret+=A[x][i].p*p;  }  return ret;}int main(){  for(scanf("%d",&T);T--;printf("%.4f\n",ans)){    scanf("%d%d",&n,&m);lim=n/2;    for(i=0;i<n;i++)scanf("%d%lf",&a[i].x,&a[i].p),a[i].p*=0.01;    for(i=1;i<=m;i++)scanf("%d%d",&b[i].c,&b[i].d);    for(i=0;i<=m;i++)for(j=0;j<=m;j++)for(k=0;k<=n;k++)f[i][j][k]=inf;    f[0][0][0]=0;    for(i=1;i<=m;i++)for(j=0;j<=m;j++)for(k=0;k<=n;k++)if(f[i-1][j][k]<inf){      up(f[i][j][k],f[i-1][j][k]);      if(k+b[i].d<=n)up(f[i][j+1][k+b[i].d],f[i-1][j][k]+b[i].c);    }    for(j=0;j<=m;j++)for(k=1;k<=n;k++)up(f[m][j][k],f[m][j][k-1]);    for(i=0;i<=n;i++)ca[i]=cb[i]=0;    dfsl(0,0,0,1);    dfsr(lim,0,0,1);    for(i=0;i<=n;i++){      if(ca[i]>1)std::sort(A[i]+1,A[i]+ca[i]+1,cmp);      if(cb[i]>1)std::sort(B[i]+1,B[i]+cb[i]+1,cmp);    }    ans=p[m+1]=0;    for(i=0;i<=n;i++)for(j=m;j;j--){      p[j]=0;      for(k=0;k<=lim&&k<=i;k++)p[j]+=cal(k,i-k,f[m][j][i]);      ans+=(p[j]-p[j+1])*j;    }  }  return 0;}

  

BZOJ4374 : Little Elephant and Boxes