4007: [JLOI2015]战争调度搜索

最新推荐文章于 2019-10-26 17:30:16 发布

ws_fqk

最新推荐文章于 2019-10-26 17:30:16 发布

阅读量500

点赞数

分类专栏： My Code

本文链接：https://blog.csdn.net/Phenix_2015/article/details/50540188

版权

My Code 专栏收录该内容

349 篇文章 1 订阅

订阅专栏

一开始算错了复杂度。。
可以用f[i][j]表示以i为根的子树中有j个平民，且fa[i]->root的状态已经确定时的最优答案，那么我们可以用大约9*2^20复杂度来解决。每次确定一个状态然后继续向下搜。注意不同的状态之间不要搞乱，即数组不要忘记清零qwq。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,ans,s[1050],f[1050][1050],a[1050][1050],b[1050][1050];
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
void dfs(int x,int cnt)
{
    if (x>=1<<n-1)
    {
        f[x][0]=f[x][1]=0;
        for (int i=x>>1;i;i>>=1)
            if (s[i]==0) f[x][1]+=a[x][i];
            else f[x][0]+=b[x][i];
        return;
    }
    s[x]=0;
    dfs(x<<1,cnt>>1); dfs(x<<1|1,cnt>>1);
    int mn=min(cnt,m);
    memset(f[x],0,4*(mn+1));
    for (int i=0;i<=mn;i++)
    {
        int l=max(0,i-(cnt>>1)),r=min(cnt>>1,i);
        for (int j=l;j<=r;j++)
            f[x][i]=max(f[x][i],f[x<<1][j]+f[x<<1|1][i-j]);
    }
    s[x]=1;
    dfs(x<<1,cnt>>1); dfs(x<<1|1,cnt>>1);
    for (int i=0;i<=mn;i++)
    {
        int l=max(0,i-(cnt>>1)),r=min(cnt>>1,i);
        for (int j=l;j<=r;j++)
            f[x][i]=max(f[x][i],f[x<<1][j]+f[x<<1|1][i-j]);
    }
}
int main()
{
    n=read(); m=read();
    for (int i=1<<n-1;i<1<<n;i++)
        for (int j=i>>1;j;j>>=1)
            a[i][j]=read();
    for (int i=1<<n-1;i<1<<n;i++)
        for (int j=i>>1;j;j>>=1)
            b[i][j]=read();
    dfs(1,1<<n-1);
    for (int i=0;i<=m;i++) ans=max(ans,f[1][i]);
    cout << ans;
    return 0;
}