hiho一下第76周《Suzhou Adventure》

  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 }