1076: [SCOI2008]奖励关

链接

思路

　　状压dp，dp[i][s]表示当前是第i次选择，吃过的宝物的状态是s，的期望得分。

　　如果一个宝物可以吃，那么可以选择吃或不吃，取最大的。

　　zz的我：50010的空间开成了500100，然后提示tle，醉了。。

代码

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<cmath>
    
using namespace std;
    
double dp[110][50010];
int b[20],w[20];
    
inline int read() {
    int x = 0,f = 1; char ch = getchar();
    for (; !isdigit(ch); ch=getchar()) if(ch=='-') f=-1;
    for (; isdigit(ch); ch=getchar()) x = x*10+ch-'0';
    return x * f;
}
    
int main() {
    int m = read(),n = read();
    for (int i=1; i<=n; ++i) {
        w[i] = read();
        int tmp = 0,a = read();
        while (a) {
            tmp |= (1<<(a-1));
            a = read();
        }
        b[i] = tmp;
    }
     
    int t = (1 << n) - 1;
    for (int i=m; i>=1; --i) {
        for (int s=t; s>=0; --s) {
            for (int k=1; k<=n; ++k) {
                if ((b[k]&s)^b[k]) dp[i][s] += dp[i+1][s];
                else dp[i][s] += max(dp[i+1][s],dp[i+1][s|(1<<(k-1))]+w[k]);
            }
            dp[i][s] /= double(n);
        }
    }
    printf("%.6lf",dp[1][0]);
    return 0;
}