JZOJ.4685. 【NOIP2016提高A组8.12】礼物

Problem

Description

夏川的生日就要到了。作为夏川形式上的男朋友，季堂打算给夏川买一些生日礼物。
商店里一共有种礼物。夏川每得到一种礼物，就会获得相应喜悦值Wi（每种礼物的喜悦值不能重复获得）。
每次，店员会按照一定的概率Pi（或者不拿出礼物），将第i种礼物拿出来。季堂每次都会将店员拿出来的礼物买下来。
众所周知，白毛切开都是黑的。所以季堂希望最后夏川的喜悦值尽可能地高。
求夏川最后最大的喜悦值是多少，并求出使夏川得到这个喜悦值，季堂的期望购买次数。

Input

第一行，一个整数N，表示有N种礼物。
接下来N行，每行一个实数Pi和正整数Wi，表示第i种礼物被拿出来的概率和可以获得喜悦值。

Output

第一行，一个整数表示可以获得的最大喜悦值。
第二行，一个实数表示获得这个喜悦值的期望购买次数，保留3位小数。

Sample Input

3
0.1 2
0.2 5
0.3 7

Sample Output

14
12.167

Data Constraint

对于10%的数据，N=1
对于30%的数据，N≤5
对于100%的数据，N≤20,0

Solution

我们发现N≤20，于是我们果断状压DP。
设状态s,表示买走了哪些物品。依题意，得方程：

$$f_s=\Sigma f_{s'}·P_i+(1-\Sigma P_i)·f_s+1， 且（s由s'转移过来，即s=s'+2^i）$$
移项，得
$$\Sigma P_i·f_s=\Sigma f_{s'}·P_i+1$$
$\Sigma P_i$表示s包含的 $p_i$的和。
整理，得
$$f_s=\frac{\Sigma f_{s'}·P_i+1}{\Sigma P_i}$$
问题解决，时间复杂度为 $O(2^n)$

Code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
double p[22],sm,f[1050001];
int _2[25],i,j,n,s;
long long w[22],sum;
int main()
{
    _2[1]=1;
    fo(i,2,21) _2[i]=_2[i-1]*2;
    scanf("%d",&n);
    for(i=1;i<=n;i++)
    {
        scanf("%lf%lld",&p[i],&w[i]);
        sum+=w[i];
    }
    fo(s,1,_2[n+1]-1)
    {
        double sm=0;
        fo(i,1,n)
            if (s & _2[i])
            {
                sm+=p[i];
                f[s]+=f[s-_2[i]]*p[i];
            }
        f[s]=(f[s]+1)/sm;
    }
    printf("%lld\n",sum);
    printf("%.3lf",f[_2[n+1]-1]);
}

——2016.8.12