码迷,mamicode.com
首页 > 其他好文 > 详细

hiho一下第76周《Suzhou Adventure》

时间:2015-12-19 16:30:19      阅读:245      评论:0      收藏:0      [点我收藏+]

标签:

我是菜鸡,我是菜鸡,我是菜鸡。。。。重要的事说三遍

算是第一次做树形dp的题吧,不太难。。

园林构成一棵树,root为1,Hi从root出发,有k个园林必须玩,每个园林游玩后会得到权值w[i],最多玩M个园林。

经过的园林必须玩,问可得到的最大权值和。

题目链接:http://hihocoder.com/problemset/problem/1104

经典的树形dp,dp[i][j],表示以i为根的子树选j个结点可得的最大权值和。

这样dp[root][num] = g[son_num][num-1] + w[root];

g[son_num][num-1]类似于对root的儿子结点的子树做背包,g[i][j] = max( g[i-1][j-p] + dp[child_id][p] );

题解不错:http://hihocoder.com/discuss/question/2743

对于必须玩的k个点,他的处理方法我没有看懂。。。。discussion里的不错。。。游玩结点i,那么其祖先结点fa[i]必要也游玩,将这些点设为must,

将must构成的子树缩成一个点,rebuild a new tree,就成了上面说的最普通的树形dp了。

技术分享
  1 #include<bits/stdc++.h>
  2 
  3 using namespace std;
  4 const int maxn = 100 + 10;
  5 int n, k, m;
  6 
  7 vector<int> G[maxn], G1[maxn];
  8 int fa[maxn], w[maxn], must[maxn], vis[maxn];
  9 int f[maxn][maxn];
 10 
 11 void init(){
 12     for( int i = 0; i < maxn; ++i ){
 13         G[i].clear();
 14         G1[i].clear();
 15     }
 16     memset( fa, -1, sizeof(fa) );
 17     memset( f, -1, sizeof(f) );
 18     memset( must, 0, sizeof(must) );
 19 }
 20 
 21 void dfs( int u, int pa ){
 22     fa[u] = pa;
 23     for( int i = 0; i < G[u].size(); ++i ){
 24         int v = G[u][i];
 25         if( v == pa )
 26             continue;
 27         dfs( v, u );
 28     }
 29 }
 30 
 31 void dfs1( int u, int pa ){
 32     if( pa != -1 ){
 33         if( vis[pa] && !vis[u] )
 34             G1[1].push_back(u);
 35         else if( !vis[pa] && !vis[u] )
 36             G1[pa].push_back(u);
 37     }
 38 
 39     for( int i = 0; i < G[u].size(); ++i ){
 40         int v = G[u][i];
 41         if( v == pa )
 42             continue;
 43         dfs1( v, u );
 44     }
 45 }
 46 
 47 int dp( int root, int  num ){
 48     if( num == 0 )
 49         return 0;
 50     if( f[root][num] != -1 )
 51         return f[root][num];
 52 
 53     int g[maxn][maxn], son_num = G1[root].size();
 54     memset( g, 0, sizeof(g) );
 55     for( int i = 1; i <= son_num; ++i ){
 56         int child = G1[root][i-1];
 57         for( int j = 0; j < num; ++j ){
 58             for( int p = 0; p <= j; ++p ){
 59                 g[i][j] = max( g[i][j], g[i-1][j-p] + dp( child, p ) );
 60             }
 61         }
 62     }
 63 
 64     f[root][num] = g[son_num][num-1] + w[root];
 65     return f[root][num];
 66 }
 67 
 68 void print( int num ){
 69     for( int i = 0; i < G1[num].size(); ++i )
 70         cout << G1[num][i] << "   ";
 71     cout << endl;
 72 }
 73 
 74 void solve(){
 75     dfs(1, -1);
 76 
 77     //must节点压缩
 78     int ans = 0, cnt = 0;
 79     memset( vis, 0, sizeof(vis) );
 80     for( int i = 1; i <= n; ++i ){
 81         if(must[i]){
 82             int u = i;
 83             while(u != -1 && !vis[u]){
 84                 vis[u] = 1;
 85                 cnt++, ans += w[u];
 86                 u = fa[u];
 87             }
 88         }
 89     }
 90     //cout << "ans: " << ans << endl;
 91 
 92     if( cnt > m ){
 93         printf("-1\n");
 94         return;
 95     }
 96 
 97     //rebuild tree
 98     dfs1( 1, -1 );
 99 
100     w[1] = ans;
101     //cout << "m-cnt: " << m - cnt << endl;
102     ans = dp( 1, m-cnt+1 );
103     //cout << "f: " << f[8][3] << endl;
104 
105     printf("%d\n", ans);
106 }
107 
108 int main(){
109     //freopen("1.in", "r", stdin);
110     init();
111     scanf("%d%d%d", &n, &k, &m);
112     for( int i = 1; i <= n; ++i ){
113         scanf("%d", &w[i]);
114     }
115 
116     int t;
117     for( int i = 1; i <= k; ++i ){
118         scanf("%d", &t);
119         must[t] = 1;
120     }
121 
122     int a, b;
123     for( int i = 1; i <= n-1; ++i ){
124         scanf("%d%d", &a, &b);
125         G[a].push_back(b), G[b].push_back(a);
126     }
127 
128     solve();
129 
130     return 0;
131 }
View Code

 

hiho一下第76周《Suzhou Adventure》

标签:

原文地址:http://www.cnblogs.com/zhazhalovecoding/p/5059242.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!