[JLOI2015]bzoj 4007 战争调度 - dp

设dp[i][j][s]表示到i这个点子树里面选了j个黑点向上的状态是s的答案，注意到(i,s)的状态数之和其实只有2的n次这么多左右，所以直接做就可以了。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<map>
#define MXS 550
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
map<int,int> dp[MXS<<1][MXS];
int main()
{
    int n,m;scanf("%d%d",&n,&m);
    for(int i=1<<(n-1);i<(1<<n);i++)
        for(int j=0;j<n-1;j++)
        {
            int x;scanf("%d",&x);
            for(int s=0;s<(1<<(n-1));s++)
                if((s>>j)&1) dp[i][1][s]+=x;
        }
    for(int i=1<<(n-1);i<(1<<n);i++)
        for(int j=0;j<n-1;j++)
        {
            int x;scanf("%d",&x);
            for(int s=0;s<(1<<(n-1));s++)
                if(!((s>>j)&1)) dp[i][0][s]+=x;
        }
    for(int x=(1<<(n-1))-1;x;x--)
    {
        int lc=x<<1,rc=lc|1,c=0,y=x;
        while(y) y>>=1,c++;
        for(int j=0;j<=min(1<<(n-c),m);j++)
            for(int s=0;s<(1<<(c-1));s++)
            {
                int v=0;
                for(int k=0;k<=j;k++)
                    v=max(v,max(dp[lc][k][s<<1]+dp[rc][j-k][s<<1],dp[lc][k][s<<1|1]+dp[rc][j-k][s<<1|1]));
                dp[x][j][s]=v;
            }
    }
    int ans=0;
    for(int i=0;i<=m;i++) ans=max(ans,dp[1][i][0]);
    return !printf("%d\n",ans);
}