hdu 5242 Game

        一棵树，点上有权值。找k条树根到叶子的路径，使权值和最大（一个点的权不会被重复计算）。

        贪心。dfs计算根到每个节点的权值和，然后对叶子的权值和排序。按顺序从大到小找叶子，对每个叶子往上累加。。然后从大到小选k条路径就可以了。

#include <queue>  
#include <stack>  
#include <math.h>  
#include <stdio.h>  
#include <stdlib.h>  
#include <iostream>  
#include <limits.h>  
#include <string.h>  
#include <string>  
#include <map>
#include <algorithm>  

using namespace std;

#define ll long long

int a[100010];
int fa[100010];
ll tot[100010];
bool vis[100010];
vector<int> E[100010];
int cnt;
ll ans[100010];

void init(){
	memset(E,0,sizeof(E));
	memset(vis,0,sizeof(vis));
	memset(ans,0,sizeof(ans));
	cnt=0;
}

struct leaf{
	int id;
	ll val;
	bool operator<(const leaf &other)const{
		return val<other.val;
	}
}Leaf[100010];

void dfs(int u){
	int sz=E[u].size();
	for(int i=0;i<sz;i++){
		int v=E[u][i];
		tot[v]=tot[u]+a[v];
		dfs(v);
		
	}
	if(!sz){
		Leaf[cnt].id=u;
		Leaf[cnt].val=tot[u];
		cnt++;
	}
}

int main(){
	int t;
	cin>>t;
	int cas=0;
	while(t--){
		cas++;
		init();
		int n,k;
		cin>>n>>k;
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
		}
		
		for(int i=1;i<n;i++){
			int u,v;
			scanf("%d%d",&u,&v);
			E[u].push_back(v);
			fa[v]=u;
		}
		
		tot[1]=a[1];
		dfs(1);
		sort(Leaf,Leaf+cnt);
		
		for(int i=cnt-1;i>=0;i--){
			int cur=Leaf[i].id;
			while(cur&&!vis[cur]){
				ans[i]+=a[cur];
				vis[cur]=1;
				cur=fa[cur];
			}
		}
		
		sort(ans,ans+cnt);
		reverse(ans,ans+cnt);
		
		ll re=0;
		for(int i=0;i<k;i++){
			re+=ans[i];
		}
		printf("Case #%d: ",cas);
		cout<<re<<endl;
	}
	return 0;
}