URAL1326. Bottle Taps(状压）

1326

用队列优化的 不知道为什么一直WA  传统直白的 状压 写了超时 O（(1<<n)*m*n) 之后想了可以把n省去 预处理一下方案

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stdlib.h>
#include<vector>
#include<cmath>
using namespace std;
#define N 1100000
#define INF 0xfffffff
int dp[2][N];
int p[25],o[110],sum[110];
vector<int>a[110];
bool f[2][N],ff[25];
int main()
{
    int i,j,n,m,v,s=0;
    cin>>n;
    for(i = 1; i <= n ; i++)
    cin>>p[i];
    cin>>m;
    for(i = 1; i <= m ;i++)
    {
        int k,b;
        cin>>o[i]>>k;
        for(j = 1; j <= k ; j++)
        {
            cin>>b;
            sum[i]+=(1<<(b-1));
        }
    }
    cin>>v;
    for(i = 1; i <= v ; i++)
    {
        int x;
        cin>>x;
        ff[x] = 1;
        s+=(1<<(x-1));
    }
    int tt[2]={0};
    tt[0]=1;
    for(i =0  ;i < (1<<n) ; i++)
    dp[0][i] = dp[1][i] = INF;
    for(i = 0; i < (1<<n)  ; i++)
    {
        int pp=0,flag=1;
        for(j = 0 ; j < n ; j++)
        {
            if(i&(1<<j))
            {
                if(!ff[j+1]) flag=0;
                pp+=p[j+1];
            }
        }
        if(!flag) continue;
        dp[0][i] = pp;
    }
    for(i = 1; i <= m ;i++)
    {
        int t1 = i%2,t2 = (i-1)%2;
        for(j = 0 ; j < (1<<n) ; j++)
        {
            dp[t1][j] = min(dp[t1][j],dp[t2][j]);
            if(dp[t2][j]==INF) continue;
            dp[t1][j|sum[i]] = min(dp[t1][j|sum[i]],dp[t2][j]+o[i]);
        }
    }
    int minz = INF;
    for(i = 0 ; i < (1<<n) ; i++)
    {
        if((s&i)==s)
        {
            minz = min(minz,min(dp[0][i],dp[1][i]));
        }
    }
    cout<<minz<<endl;
    return 0;
}