HDU 5242 Game(树上贪心)

题目链接 Game

题目的意思很简单， 就是要找一棵树权值最大等等前K条链。

在本题中，走的次数等于min(叶子结点个数，k)

tree[i].sum意为从i号结点出发走到某个叶子结点能得到的最大总价值。

pson[i]表示i号结点若要获得最大价值那么下一步该怎么走。

显然tree[i].sum和pson[i]是从i的各个儿子转移得到的。

那么先做一遍DFS计算出tree[i].sum， 再排序。

然后贪心，从价值最大的那个结点开始选，从大到小。

选的时候，要把他中途经过的结点全部屏蔽（就是说被屏蔽的结点接下来不能被选了）。

当选的次数到达k或者没有结点可选时那就停止。

具体可以看我的代码。

 #include <bits/stdc++.h>

 using namespace std;

 #define REP(i,n)                for(int i(0); i <  (n); ++i)
 #define rep(i,a,b)              for(int i(a); i <= (b); ++i)
 #define dec(i,a,b)              for(int i(a); i >= (b); --i)
 #define for_edge(i,x)           for(int i = H[x]; i; i = X[i])

 #define LL      long long

 const int N     =          +       ;
 const int M     =           +       ;
 const int A     =              +       ;

 struct Node{
     LL sum; int id;
     friend bool operator < (const Node &a, Node &b){
         return a.sum > b.sum;
     }
 } tree[N];

 int E[N << ], X[N << ], H[N << ], pson[N];
 bool isleaf[N], v[N], vc[N];
 LL a[N], c[N], ans;
 int T, n, k, et, x, y, num;
 int Case = ;
 int cnt = ;

 bool cmp(LL a, LL b){ return a > b;}

 inline void addedge(int a, int b){
     E[++et] = b, X[et] = H[a], H[a] = et;
 }

 void dfs(int x, int fa){
     LL cnt = , tot = , w = ;
     tree[x].sum = a[x];
     for_edge(i, x) if (E[i] != fa){
         int u = E[i]; ++cnt;
         dfs(u, x);
         if (tree[u].sum > tot){ tot = tree[u].sum; w = u;}
     }
     pson[x] = w;
     if (cnt == )  isleaf[x] = true;
     else tree[x].sum += tree[w].sum;
 }

 void tag(int x){
     v[x] = false;
     if (!isleaf[x]) tag(pson[x]);
 }

 int main(){

     scanf("%d", &T);
     while (T--){
         et = ; num = ;
         scanf("%d%d", &n, &k);
         memset(H, , sizeof H);
         memset(c, , sizeof c);
         memset(pson, , sizeof pson);
         memset(vc, false, sizeof vc);
         memset(isleaf, false, sizeof isleaf);
         rep(i, , n) scanf("%lld", a + i);
         rep(i, , n - ){
             scanf("%d%d", &x, &y);
             addedge(x, y); addedge(y, x);
             vc[y] = true;
         }
         rep(i, , n) tree[i].id = i;
         dfs(, );
         sort(tree + , tree + n + );
         memset(v, true, sizeof v);
         cnt = ; ans = ;
         rep(i, , n) if (v[tree[i].id] && cnt < k){
             ans += tree[i].sum;
             tag(tree[i].id);
             ++cnt;
         }

         printf("Case #%d: %lld\n", ++Case, ans);

     }

     return ;

 }