HDU 5242 Game（贪心）

http://acm.hdu.edu.cn/showproblem.php?pid=5242

题意：

给出一棵树，每个节点都有一个权值，每次可以获得从根结点（1）到叶子节点上的所有权值和，每个节点只能获得一次。求k次操作后可以获得的最大权值和。

思路：

反向建图，首先求出所有节点到根节点的权值和，然后降序排序，从可以获得最大权值和的叶子节点开始计算。已经遍历过的节点就直接返回0。计算完后再次重新排序，取前面的k个较大值。

 #include<iostream>
 #include<algorithm>
 #include<cstdio>
 #include<cstring>
 using namespace std;
 const int maxn = 1e5+;
 typedef long long ll;
 int tot, n, k;
 ll val[maxn],head[maxn];
 bool vis[maxn];

 struct node
 {
     int v,next;
 }e[maxn];

 struct Node
 {
     int id;
     ll sum;
     bool operator< (const Node& rhs)const
     {
         return sum>rhs.sum;
     }
 }t[maxn];

 void addEdge(int u, int v)
 {
     e[tot].v = v;
     e[tot].next = head[u];
     head[u] = tot++;
 }

 ll dfs1(int u)
 {
     if(vis[u])  return t[u].sum;
     vis[u] = true;
     t[u].sum = val[u];
     for(int i=head[u];i!=-;i=e[i].next)
     {
         int v = e[i].v;
         t[u].sum += dfs1(v);
     }
     return t[u].sum;
 }

 ll dfs2(int u)
 {
     if(vis[u])  return ;
     vis[u] = true;
     ll tmp = val[u];
     for(int i=head[u];i!=-;i=e[i].next)
     {
         int v = e[i].v;
         tmp += dfs2(v);
     }
     return tmp;
 }

 int main()
 {
     //freopen("in.txt","r",stdin);
     int T;
     int cas = ;
     scanf("%d",&T);
     while(T--)
     {
         tot = ;
         memset(head,-,sizeof(head));
         scanf("%d%d",&n,&k);
         for(int i=;i<=n;i++)  scanf("%lld",&val[i]);
         for(int i=;i<n;i++)
         {
             int u,v;
             scanf("%d%d",&u,&v);
             addEdge(v,u);
         }
         memset(vis,,sizeof(vis));
         memset(t,,sizeof(t));
         for(int i=;i<=n;i++)
         {
             t[i].id = i;
             t[i].sum = dfs1(i);
         }
         sort(t+,t+n+);
         memset(vis,,sizeof(vis));
         for(int i=;i<=n;i++)
         {
             int root = t[i].id;
             t[i].sum = dfs2(root);
         }
         sort(t+,t+n+);
         ll ans = ;
         for(int i=;i<=k;i++)
             ans += t[i].sum;
         printf("Case #%d: %lld\n",++cas,ans);
     }
     return ;
 }