树形背包

原题为HDU1561 The more,The better

Problem Description
ACboy很喜欢玩一种战略游戏,在一个地图上,有N座城堡,每座城堡都有一定的宝物,在每次游戏中ACboy允许攻克M个城堡并获得里面的宝物。但由于地理位置原因,有些城堡不能直接攻克,要攻克这些城堡必须先攻克其他某一个特定的城堡。你能帮ACboy算出要获得尽量多的宝物应该攻克哪M个城堡吗?
 
Input
每个测试实例首先包括2个整数,N,M.(1 <= M <= N <= 200);在接下来的N行里,每行包括2个整数,a,b. 在第 i 行,a 代表要攻克第 i 个城堡必须先攻克第 a 个城堡,如果 a = 0 则代表可以直接攻克第 i 个城堡。b 代表第 i 个城堡的宝物数量, b >= 0。当N = 0, M = 0输入结束。
 
Output
对于每个测试实例,输出一个整数,代表ACboy攻克M个城堡所获得的最多宝物的数量。
 
Sample Input
3 2
0 1
0 2
0 3
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2
0 0
 
Sample Output
5
13
这道题有两种算法,第一种算法是O(nw^2)的,就是直接在树上dp,之后枚举每一个节点,枚举每一个节点取得最大体积和当前取得的体积,就可以。
不过我们今天来说一下O(nw)的算法,因为普通的背包是可以线性解题的,这样才能达到二维的时间复杂度,那么我们怎么把一棵树转化为线性问题求解呢?
我们可以先求这棵树的dfs序,之后记录每一个点的子树大小,之后像正常dp一样讨论取或不取即可。
if(j+1<=m) dp[i+1][j+1] = max(dp[i+1][j+1],dp[i][j]+w[seq[i]]);
            dp[i+size[seq[i]]][j] = max(dp[i+size[seq[i]]][j],dp[i][j]);

第一行是取,第二行是不取,其中dp[i][j]表示在dfs序中到第i个节点已经用了j个体积所能获得的最大价值。因为这道题攻陷每个城堡所用的消耗固定为1,所以直接+1就可以。

不过这题并不是直接照抄dp转移方程就可以!这里有好多需要注意的事情。

第一个,我们在树形dp转化为线性问题的时候,要注意一个状态是不可以凭空生成的,也就是说我们不能在没有取父节点的情况下直接选取子节点,这会对第一行代码产生极大的影响,所以我们开一个vis数组以记录这种状态是否被取过,如果被取过的话我们就可以继续dp,否则不可以。

第二个,注意子树大小是怎么计算的……他是算自己的。所以我们递归的时候应该先把每个节点初始化为1,之后(我用的是链前)直接把其所有子节点的子树大小加起来就可以了。

第三个是dfs序的问题,这里一定要注意i表示dfs序中到第i个点,所以dfs序从0还是1开始存将会有至关重要的影响。在代码里我是从1开始存并从1开始dp,所以最后我们只要去dp[tot][m]就可以,其他情况需要视情况而定,如果实在不行的话自己打表分析一下。

这样就可以过了……要注意的东西挺多的,而且HDU不提供数据点debug很难受……

最后上代码。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<cmath>
#include<queue>
#define rep(i,a,n) for(int i = a;i <= n;i++)
#define per(i,n,a) for(int i = n;i >= a;i--)
#define enter putchar('\n')
#define M 2005
using namespace std;
typedef long long ll;
const int INF = 2147483647;
struct node
{
    int next,to;
}edge[M];
int n,dp[M][M],head[M],size[M],cnt,tot,ans,m,seq[M],w[M];
bool vis1[M][M];
int read()
{
    int ans = 0,op = 1;
    char ch = getchar();
    while(ch < '0' || ch > '9')
    {
        if(ch == '-') op = -1;
        ch = getchar();
    }
    while(ch >= '0' && ch <= '9')
    {
        ans *= 10;
        ans += ch - '0';
        ch = getchar();
    }
    return ans * op;
}
void init()
{
    memset(edge,0,sizeof(edge));
    memset(head,0,sizeof(head));
    memset(dp,0,sizeof(dp));
    memset(size,0,sizeof(size));
    memset(seq,0,sizeof(seq));
    memset(w,0,sizeof(w));
    memset(vis1, 0, sizeof(vis1));
    cnt = 0,tot = 0;
}
void add(int x,int y)
{
    edge[++cnt].next = head[x];
    edge[cnt].to = y;
    head[x] = cnt;
}
void build()
{
    rep(i,1,n)
    {
        int x = read();
        int y = read();
        add(x,i);
        w[i] = y;
    }
}
void dfs(int x)
{
    seq[tot++] = x; size[x] = 1;
    for(int i = head[x];i;i = edge[i].next)
    {
        int v = edge[i].to;
        dfs(v);
        size[x] += size[v];
    }
} 
void DP()
{
    vis1[1][0] = 1;
    rep(i,1,tot + 1)
    rep(j,0,n)
    {
        if(vis1[i][j])
        {
            if(j+1<=m) dp[i+1][j+1] = max(dp[i+1][j+1],dp[i][j]+w[seq[i]]);
            dp[i+size[seq[i]]][j] = max(dp[i+size[seq[i]]][j],dp[i][j]);
            vis1[i+1][j+1] = vis1[i+size[seq[i]]][j] = 1;
        }
    }
}
int main()
{
    while(1)
    {
        n = read(),m = read();
        if(!n && !m) break;
        init();
        build();
        dfs(0);
        DP();
        printf("%d\n",dp[tot][m]);
    }
    return 0;
}

 

posted @ 2018-04-20 00:36  CaptainLi  阅读(176)  评论(0编辑  收藏  举报