【Luogu】P2465山贼集团（树形状压DP）

　　题目链接

　　写了个70分暴力还挂了，第一遍提交只拿了十分……海星

　　首先建虚拟节点多叉树转成二叉，然后子集枚举DP

　　设g[x][i]是以x为根的子树内山贼集合i，x啥都不选也没贡献的时候的最大价值

　　f[x][i]是要求的答案

　　然后状压DP即可。

#include<cstdio>
#include<algorithm>
#include<cctype>
#include<cstring>
#include<cstdlib>
#define maxn 200
#define maxp 13
using namespace std;
inline long long read(){
    long long num=,f=;
    char ch=getchar();
    while(!isdigit(ch)){
        if(ch=='-')    f=-;
        ch=getchar();
    }
    while(isdigit(ch)){
        num=num*+ch-'';
        ch=getchar();
    }
    return num*f;
}

struct Edge{
    int next,to;
};

struct Pic{
    Edge edge[maxn*];
    int head[maxn*],num;
    Pic(){num=;}
    void add(int from,int to){
        edge[++num]=(Edge){head[from],to};
        head[from]=num;
    }
}Old,New;

int cst[maxn][maxn];
int tot;
int mon[<<maxp];
int dl[maxn],ndl[maxn*];
int f[maxn*][<<maxp];
int g[maxn*][<<maxp];
int cost[maxn][<<maxp];
int vl[<<maxp];
int n,p;

void chan(int x,int fa){
    int now=x;
    for(int i=Old.head[x];i;i=Old.edge[i].next){
        int to=Old.edge[i].to;
        if(to==fa)    continue;
        if(ndl[now]==||(ndl[now]==&&dl[x]==)){
            New.add(now,to);
            ndl[now]++;    dl[x]--;
        }
        else{
            New.add(now,++tot);
            now=tot;
            New.add(now,to);
            ndl[now]++;    dl[x]--;
        }
        chan(to,x);
    }
}

void dfs(int x,int fa){
    int lson=,rson=;
    f[x][]=;
    g[x][]=;
    for(int i=New.head[x];i;i=New.edge[i].next){
        int to=New.edge[i].to;
        if(to==fa)    continue;
        if(lson==)    lson=to;
        else        rson=to;
        dfs(to,x);
    }
    if(x>n){
        if(rson==){
            for(int i=;i<(<<p);++i)    f[x][i]=g[x][i]=f[lson][i];
            return;
        }
        for(int i=;i<(<<p);++i){
            for(int j=i;j;j=(j-)&i)
                    g[x][i]=max(g[x][i],f[lson][j]+f[rson][i^j]);
            g[x][i]=max(g[x][i],f[lson][]+f[rson][i]);
                f[x][i]=g[x][i];
            }
      }
    else{
        if(rson==){
            for(int i=;i<(<<p);++i){
                g[x][i]=f[lson][i];
                f[x][i]=g[x][i]+mon[i];
            }
            for(int i=;i<(<<p);++i)
                for(int j=i;j;j=(j-)&i)
                    f[x][i]=max(f[x][i],g[x][i^j]-cost[x][j]+mon[i]);
            return;
        }
        for(int i=;i<(<<p);++i){
            for(int j=i;j;j=(j-)&i)
                    g[x][i]=max(g[x][i],f[lson][j]+f[rson][i^j]);
            g[x][i]=max(g[x][i],f[lson][]+f[rson][i]);
                f[x][i]=g[x][i]+mon[i];
            }
        for(int i=;i<(<<p);++i)
            for(int j=i;j;j=(j-)&i){
                int costx=cost[x][j],state=i^j;
                f[x][i]=max(f[x][i],g[x][state]+mon[i]-costx);
            }

    }
}

int main(){
    memset(f,-/,sizeof(f));
    memset(g,-/,sizeof(g));
    g[][]=;
    f[][]=;
    n=read(),p=read();
    tot=n;
    for(int i=;i<n;++i){
        int from=read(),to=read();
        Old.add(from,to);
        Old.add(to,from);
        dl[from]++;    dl[to]++;
    }
    for(int i=;i<=n;++i)    dl[i]--;
    for(int i=;i<=n;++i)
        for(int j=;j<=p;++j)    cst[i][j]=read();
    int T=read();
    while(T--){
        int val=read(),cnt=read(),state=;
        for(int i=;i<=cnt;++i){
            int x=read();
            state|=(<<(x-));
        }
        vl[state]+=val;
    }
    for(int i=;i<(<<p);++i)
        for(int j=i;j;j=(j-)&i)    mon[i]+=vl[j];
    for(int i=;i<=n;++i)
        for(int j=;j<(<<p);++j){
            int costx=;
            for(int k=;k<=p;++k)
                if(j&(<<(k-)))    costx+=cst[i][k];
            cost[i][j]=costx;
        }
    chan(,);
    dfs(,);
    printf("%d\n",f[][(<<p)-]);
}
/*
9 3

1 2
1 3
1 4
2 5
3 6
3 7
3 8
6 9

*/