[好题][CodeChef]LEBOXES/[JZOJ4703]Buy

题目大意

有$n$个钱袋，每个钱袋有$P_i$的概率是价值$V_i$的钱币，有$1-P_i$的概率是$1$颗钻石。
有$m$个食品，每个食品购买要支付$C_i$的钱币和$D_i$的钻石。
求最多购买食品数量的期望。
一个测试点$T$组数据。

$1\le n,m\le 30,T\le5,V_i,C_i\le10^7,1\le P_i\le10^2$

---

题目分析

首先思考一个最暴力的算法。枚举每个钱袋的状态，然后做动态规划。
令$f_{i,j,k}$表示考虑到第$i$个食物，消耗价值为$j$的钱币以及$k$颗钻石，能购买的最多食品数量。
时间复杂度$\mathrm O(2^nmn^2V)$。
可以发现上面的动态规划和钱袋的状态无关，我们先考虑如何将它改写为一个具有启发性（雾？）的形式。由于每个钱袋最多只会提供一颗钻石，因此钻石数量是$\mathrm O(n)$的，但是钱币数量可能会很大，将它放入$dp$状态中是很不明智的。
我们考虑转化一下，设$f_{i,j}$表示在消耗了$j$个钻石的情况下总共买$i$个食品消耗的最小钱币数。这样就可以避免状态的巨大消耗。
那么我们考虑怎么优化枚举钱袋状态的复杂度。$n\le30$启发我们去使用折半搜索（$meet\ in\ the\ middle$）。
考虑将钱袋分成前后两个部分，大小分别为$a$和$b$。
对于前面的部分，我们可以暴力枚举状态$(P,V,D)$表示概率/钱币/钻石。先不讨论应该如何储存。
对于后面的部分，我们依然暴力枚举状态$(P,V,D)$，然后考虑与前面的合并。注意到钻石数和食品数都是$\mathrm O(n)$的可枚举数。我们枚举最后买了食品数量$i$，枚举前半部分使用钻石数量$d$，那么前半部分的消耗钱数$G$应该满足$f_{i,d+D}\le V+G\lt f_{i+1,d+D}$。不能太少也不能太多。
注意到这里我们确定了前半部的钻石数和钱币的取值区间，我们要统计符合条件的状态的概率和。于是在处理前半部分的时候，我们可以将状态按照钻石数分组，然后组内按照钱币数排序，预处理概率的前缀和。查询的时候二分区间即可。
前半部分复杂度$\mathrm O(2^a\log_22^a)=\mathrm O(a2^a)$，后半部分复杂度$\mathrm O(2^b\times m\times a\times\log_22^a)=\mathrm O(a^22^bm)$。在这里$a$取$\frac{2}{3}n$，$b$取$\frac{1}{3}n$即可通过所有数据。当然当$a$超过$18$的时候空间需求可能很大，我们可以适当调整一下 a <script type="math/tex" id="MathJax-Element-40">a</script>。
还有什么不懂就参考代码实现吧。

---

代码实现

#include <algorithm>
#include <iostream>
#include <cstring>
#include <climits>
#include <cstdio>
#include <cctype>
#include <cmath>

using namespace std;

typedef double db;

typedef pair<int,db> PI;
#define mkp(a,b) make_pair(a,b)
#define ft first
#define sd second

const int N=35;
const int M=35;
const int A=18;
const int S=1<<A;
const int INF=INT_MAX/3;

int V[N],P[N],C[M],D[M];
int n,m,T,a,b;
PI sta[A+5][S];
db sum[A+5][S];
int f[N][M];
int cnt[A+5];
db ans;

void dp()
{
    for (int i=0;i<=n;i++)
        for (int j=0;j<=m+1;j++)
            f[i][j]=INF;
    for (int i=0;i<=n;i++) f[i][0]=0;
    for (int i=1;i<=m;i++)//food id
        for (int k=i;k>=1;k--)//food sum
            for (int j=n;j>=D[i];j--)//diamond
                f[j][k]=min(f[j][k],f[j-D[i]][k-1]+C[i]);
}

void store(db pro,int d,int v){sta[d][++cnt[d]]=mkp(v,pro);}

void dfs1(int x,db pro,int d,int v)//id probably diamonds value
{
    if (x==a+1)
    {
        store(pro,d,v);
        return;
    }
    dfs1(x+1,pro*P[x]/100.0,d,v+V[x]),dfs1(x+1,pro*(100-P[x])/100.0,d+1,v);
}

void make_pre()
{
    dfs1(1,1.0,0,0);
    for (int i=0;i<=a;i++)
        if (cnt[i])
        {
            sort(sta[i]+1,sta[i]+cnt[i]+1),sum[i][0]=0;
            for (int j=1;j<=cnt[i];j++) sum[i][j]=sum[i][j-1]+sta[i][j].sd;
        }
}

int search(int x,int d)
{
    int ret=0,l=1,r=cnt[d],mid;
    while (l<=r)
    {
        mid=l+r>>1;
        if (sta[d][mid].ft<x) ret=mid,l=mid+1;
        else r=mid-1;
    }
    return ret;
}

void calc(db pro,int d,int v)
{
    for (int d0=0;d0<=a&&d0<=n-d;d0++)
    {
        if (!cnt[d0]) continue;
        for (int ts=1;ts<=m;ts++)
        {
            int L=f[d0+d][ts]-v,R=f[d0+d][ts+1]-v;
            int l=search(L,d0),r=search(R,d0);
            ans+=ts*pro*(sum[d0][r]-sum[d0][l]);
        }
    }
}

void dfs2(int x,db pro,int d,int v)
{
    if (x==n+1)
    {
        calc(pro,d,v);
        return;
    }
    dfs2(x+1,pro*P[x]/100.0,d,v+V[x]),dfs2(x+1,pro*(100-P[x])/100.0,d+1,v);
}

void solve()
{
    ans=0.0,dfs2(a+1,1.0,0,0);
}

int main()
{
    freopen("buy.in","r",stdin),freopen("buy.out","w",stdout);
    for (scanf("%d",&T);T;T--)
    {
        scanf("%d%d",&n,&m);
        memset(cnt,0,sizeof cnt),memset(sta,0,sizeof sta),memset(sum,0,sizeof sum);
        for (int i=1;i<=n;i++) scanf("%d%d",&V[i],&P[i]);
        for (int i=1;i<=m;i++) scanf("%d%d",&C[i],&D[i]);
        a=min(n*2/3,A),b=n-a;
        dp(),make_pre(),solve();
        printf("%.4lf\n",ans);
    }
    fclose(stdin),fclose(stdout);
    return 0;
}