POJ 2486 Apple Tree 树形dp
2012-04-03 23:54
405 查看
http://poj.org/problem?id=2486
题意:一个叫Wshxzt的可爱的女孩子被HX大叔带到了一棵苹果树边。众所周知,苹果树是一个树形的结构,在节点处长有苹果(这明显不符合实际情况……)。现在我们知道Wshxzt是个苹果控,她只要访问到一个节点,就一定会吃光这个节点所有的苹果。HX大叔为了防止Wshzxt长胖,限制她只能走K(1 ≤ K ≤ 200)步,从一个节点走到另一个相邻的节点是所谓走一步。Wshxzt从节点1开始。树上的节点有N(1 ≤ N ≤ 100)个,你需要计算Wshxzt最多能吃到多少苹果。
思路:树形dp,比较经典的一个树形dp。首先很容易就可以想到用dp[root][k]表示以root为根的子树中最多走k时所能获得的最多苹果数,接下去我们很习惯地会想到将k步在root的所有子结点中分配,也就是进行一次背包,就可以得出此时状态的最优解了,但是这里还有一个问题,那就是在进行背包的时候,对于某个孩子son走完之后是否回到根结点会对后面是否还能分配有影响,为了解决这个问题,我们只需要在状态中增加一维就可以了,用dp[root][k][0]表示在子树root中最多走k步,最后还是回到root处的最大值,dp[root][k][1]表示在子树root中最多走k步,最后不回到root处的最大值。由此就可以得出状态转移方程了:
dp[root][j][0] = MAX (dp[root][j][0] , dp[root][j-k][0] + dp[v][k-2][0]);
dp[root][j]][1] = MAX( dp[root][j][1] , dp[root][j-k][0] + dp[v][k-1][1]) ;
dp[root][j][1] = MAX (dp[root][j][1] , dp[root][j-k][1] + dp[v][k-2][0]);
代码:
题意:一个叫Wshxzt的可爱的女孩子被HX大叔带到了一棵苹果树边。众所周知,苹果树是一个树形的结构,在节点处长有苹果(这明显不符合实际情况……)。现在我们知道Wshxzt是个苹果控,她只要访问到一个节点,就一定会吃光这个节点所有的苹果。HX大叔为了防止Wshzxt长胖,限制她只能走K(1 ≤ K ≤ 200)步,从一个节点走到另一个相邻的节点是所谓走一步。Wshxzt从节点1开始。树上的节点有N(1 ≤ N ≤ 100)个,你需要计算Wshxzt最多能吃到多少苹果。
思路:树形dp,比较经典的一个树形dp。首先很容易就可以想到用dp[root][k]表示以root为根的子树中最多走k时所能获得的最多苹果数,接下去我们很习惯地会想到将k步在root的所有子结点中分配,也就是进行一次背包,就可以得出此时状态的最优解了,但是这里还有一个问题,那就是在进行背包的时候,对于某个孩子son走完之后是否回到根结点会对后面是否还能分配有影响,为了解决这个问题,我们只需要在状态中增加一维就可以了,用dp[root][k][0]表示在子树root中最多走k步,最后还是回到root处的最大值,dp[root][k][1]表示在子树root中最多走k步,最后不回到root处的最大值。由此就可以得出状态转移方程了:
dp[root][j][0] = MAX (dp[root][j][0] , dp[root][j-k][0] + dp[v][k-2][0]);
dp[root][j]][1] = MAX( dp[root][j][1] , dp[root][j-k][0] + dp[v][k-1][1]) ;
dp[root][j][1] = MAX (dp[root][j][1] , dp[root][j-k][1] + dp[v][k-2][0]);
代码:
#include<stdio.h> #include<string.h> #define MAX(a,b) a>b?a:b int N ,K ; int value[110] ; bool maze[110][110] ; int f[110] ; int dp[110][210][2] ; void Build(int u){ for(int v = 1 ;v<=N;v++){ if(maze[u][v]==0 || v==f[u]) continue ; f[v] = u ; Build(v) ; } } void dfs(int u){ //以u为根的子树中,最多走k步能得出的最多苹果数 for(int i=0;i<=K;i++){ dp[u][i][1] = value[u] ; dp[u][i][0] = value[u] ; } for(int v=1;v<=N;v++){ if(maze[u][v]==0 || v==f[u]) continue ; dfs(v); for(int j=K ;j>=0;j--){ for(int k=1;j-k>=0;k++){ dp[u][j][1] = MAX(dp[u][j][1] , dp[u][j-k][0]+dp[v][k-1][1] ); //不回到u if(k % 2 == 0){ dp[u][j][1] = MAX(dp[u][j][1] , dp[u][j-k][1]+dp[v][k-2][0]); //注意这里,不回到u有两种情况,而回到u 只有一种情况 dp[u][j][0] = MAX(dp[u][j][0] , dp[u][j-k][0]+dp[v][k-2][0]); } } } } } int main(){ int a ,b; while(scanf("%d %d",&N,&K) == 2){ for(int i=1;i<=N;i++){ scanf("%d",&value[i]); } memset(maze, 0 , sizeof(maze )); memset(f , -1 ,sizeof(f)) ; for(int i=1;i<N;i++){ scanf("%d %d",&a,&b); maze[a][b] = maze[b][a] = 1 ; } Build(1) ; memset(dp , 0, sizeof(dp)); dfs(1); int res = MAX(dp[1][K][0] , dp[1][K][1] ); printf("%d\n",res); } return 0 ; }
相关文章推荐
- poj 2486 Apple Tree(树形dp)
- POJ 2486 Apple Tree(树形dp)
- POJ 2486 Apple Tree - 树形DP
- POJ 2486 Apple Tree 树形DP
- POJ 2486 Apple Tree(树形DP)
- POJ-2486 Apple Tree 树形DP
- poj 2486-Apple Tree - 树形DP
- poj 2486 Apple Tree(树形dp)
- POJ 2486 Apple Tree(树形dp)
- POJ 2486 Apple Tree (树形dp)
- POJ2486 - Apple Tree(树形DP)
- POJ 2486 Apple Tree 树形DP
- POJ-2486 Apple Tree (树形DP)
- poj 2486 apple tree(树形DP)
- POJ 2486-Apple Tree(树形DP)(难)
- poj 2486 Apple Tree (带回溯的树形dp)
- [树形dp] poj 2486 Apple Tree
- POJ 2486 Apple Tree 树形DP
- 树形DP______Apple Tree( POJ 2486 )
- poj 2486 Apple Tree(树形DP 状态方程有点难想)