zoj 3201 Tree of Tree（树形背包dp）

最新推荐文章于 2020-10-29 20:08:48 发布

shuangde800

最新推荐文章于 2020-10-29 20:08:48 发布

阅读量2.1k

点赞数 2

分类专栏：动态规划 ACM-ICPC征途文章标签： zoj 树形dp 背包

本文链接：https://blog.csdn.net/shuangde800/article/details/10781015

版权

动态规划同时被 2 个专栏收录

91 篇文章 2 订阅

订阅专栏

ACM-ICPC征途

60 篇文章 45 订阅

订阅专栏

本文出自 http://blog.csdn.net/shuangde800

--------------------------------------------------------------------------------------

题目链接： zoj-3201

题意

给一棵节点带权的树，找到一个有k个节点的子树，求这个子树的最大权值

思路

树形dp+背包。
f(i, j) 表示以i为根节点的有j个节点子树的最大权值
然后对i的每个子节点做分组背包，因为对于i的每个儿子，可以选择分配
1,2,3...j-1个节点给它

f(i, j) = max{ max{f(i, j-p) + f(v, p) | 1<=p<j} | v是i的儿子节点}

ans = max{ f[i][k] | 0<=i<n && i子树节点个数>=k }

代码

/**=====================================================
 *   This is a solution for ACM/ICPC problem
 *
 *   @source      : zoj-3201 Tree of Tree
 *   @description : 树形背包dp
 *   @author      : shuangde
 *   @blog        : blog.csdn.net/shuangde800
 *   @email       : zengshuangde@gmail.com
 *   Copyright (C) 2013/08/31 20:16 All rights reserved. 
 *======================================================*/
#include 
   
   
    
    
#include 
    
    
     
     
#include 
     
     
      
      
#include 
      
      
       
       
#include 
       
       
         #include 
        
          #include 
         
           #include 
          
            #include 
            #include 
            
              #define MP make_pair using namespace std; typedef pair 
             
               PII; typedef long long int64; const double PI = acos(-1.0); const int INF = 0x3f3f3f3f; const int MAXN = 110; int n, m; int weight[MAXN]; int tot[MAXN]; vector 
              
                adj[MAXN]; int f[MAXN][MAXN]; int ans; int dfs(int u, int fa) { tot[u] = 1; for (int e = 0; e < adj[u].size(); ++e) { int v = adj[u][e]; if (v == fa) continue; tot[u] += dfs(v, u); } f[u][1] = weight[u]; for (int e = 0; e < adj[u].size(); ++e) { int v = adj[u][e]; if (v == fa) continue; for (int i = tot[u]; i >= 1; --i) { for (int j = 0; j < i && j <= tot[v]; ++j) { f[u][i] = max(f[u][i], f[u][i-j] + f[v][j]); } } } if (tot[u] >= m) ans = max(ans, f[u][m]); return tot[u]; } int main(){ while (~scanf("%d%d", &n, &m)) { for (int i = 0; i < n; ++i) { scanf("%d", &weight[i]); adj[i].clear(); } for (int i = 0; i < n - 1; ++i) { int u, v; scanf("%d%d", &u, &v); adj[u].push_back(v); adj[v].push_back(u); } memset(f, 0, sizeof(f)); ans = 0; dfs(0, -1); printf("%d\n", ans); } return 0; }

shuangde800

关注

2
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
zoj 3201 Tree of Tree（树形背包dp）

题意给一棵节点带权的树，找到一个有k个节点的子树，求这个子树的最大权值思路树形dp+背包。 f(i, j) 表示以i为根节点的有j个节点子树的最大权值然后对i的每个子节点做分组背包，因为对于i的每个儿子，可以选择分配 1,2,3...j-1个节点给它 f(i, j) = max{ max{f(i, j-p) + f(v, p) | 1<=p<j} | v是i的儿子节点} ans = max{ f[i][k] | 0<=i<n && i子树节点个数>=k
复制链接

扫一扫