Luogu P3262 [JLOI2015]战争调度

题意

给定一棵高度为 \(n\) 的完全二叉树，可以将节点设置成两种状态。如果某个叶子 \(x\) 的状态为 \(i\) 同时他的某个祖先也为 \(i\)，那么这个叶子就会对祖先产生 \(f_{x,i}\) 的贡献。求叶子状态为 \(0\) 的数量小于等于 \(m\) 的最大贡献。

\(\texttt{Data Range:}1\leq n\leq 10,m\leq 2^{n-1}\)

题解

考虑先设一个 \(f_{i,j}\) 表示到了 \(i\) 点，叶子选了 \(j\) 个 \(0\) 的，这个子树对祖先的总贡献，但是会发现这个东西好像做不了。

于是考虑状压。设 \(f_{i,j,S}\) 表示到了 \(i\) 点，叶子选了 \(j\) 个 \(0\)，这个点的祖先的选择的状态为 \(S\) 时这个字数对祖先的总贡献。

然后这个东西就做个树形背包就好了。

实现的时候可以考虑爆搜一下，就可以压缩掉 \(S\) 这一维。

代码

#include<bits/stdc++.h>
using namespace std;
typedef int ll;
typedef long long int li;
const ll MAXN=2051;
ll n,m,sz,res;
ll u[MAXN][15],v[MAXN][15],f[MAXN][MAXN<<1];
inline ll read()
{
    register ll num=0,neg=1;
    register char ch=getchar();
    while(!isdigit(ch)&&ch!='-')
    {
        ch=getchar();
    }
    if(ch=='-')
    {
        neg=-1;
        ch=getchar();
    }
    while(isdigit(ch))
    {
        num=(num<<3)+(num<<1)+(ch-'0');
        ch=getchar();
    }
    return num*neg;
}
#define ls node<<1
#define rs node<<1|1
inline void dfs(ll node,ll sz,ll st)
{
    for(register int i=0;i<=sz;i++)
    {
        f[node][i]=0;
    }
    if(sz==1)
    {
        for(register int i=0;i<n-1;i++)
        {
            if(st&(1<<i))
            {
                f[node][1]+=u[node][i+1];
            }
            else
            {
                f[node][0]+=v[node][i+1];
            }
        }
        return;
    }
    for(register int i=0;i<2;i++)
    {
        dfs(ls,sz>>1,st<<1|i),dfs(rs,sz>>1,st<<1|i);
        for(register int j=0;j<=min(sz,m);j++)
        {
            for(register int k=0;k<=min(sz,m);k++)
            {
                f[node][j+k]=max(f[node][j+k],f[ls][j]+f[rs][k]);
            }
        }
    }
}
int main()
{
    n=read(),m=read(),sz=1<<n;
    for(register int i=sz>>1;i<sz;i++)
    {
        for(register int j=1;j<=n-1;j++)
        {
            u[i][j]=read();
        }
    }
    for(register int i=sz>>1;i<sz;i++)
    {
        for(register int j=1;j<=n-1;j++)
        {
            v[i][j]=read();
        }
    }
    dfs(1,sz-1,0);
    for(register int i=0;i<=m;i++)
    {
        res=max(res,f[1][i]);
    }
    printf("%d\n",res);
}