HDU5242

发表于 | 分类于 ，

HDU5242

题意

​ 一棵树有n个结点，n-1条边，每个结点有个权值。每次可以获得从根节点走到叶子结点所有结点的权值和，但是每个结点的权值只能使用一次。求走k次所能获得的最大权值和。

思路

​ 先进行第一次dfs，求出每条从根到叶子节点的链，然后将这些链排序，在进行第二遍dfs,求出抛去重复走的路径，在一次对所求的结果进行排序，取前m大加起来就行。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
typedef long long ll;
int n,m;
int head[maxn],ver[maxn],nex[maxn];
int tot,p;
int w[maxn];
int visit[maxn],fa[maxn];
ll ans[maxn];
struct point{
    ll x,y;
}a[maxn];
bool cmp(point a,point b){
    return a.y>b.y;
}
void add(int u,int v){
    ver[++tot]=v,nex[tot]=head[u],head[u]=tot;
}
void init(){
    memset(head,0,sizeof head);
    memset(visit,0,sizeof visit);
    tot=0;
    p=0;
}
void dfs(ll now,ll val){
    if(head[now]){
        for(int i=head[now];i;i=nex[i]){
            int y=ver[i];
            dfs(y,val+w[now]);
        }
    } 
    else {
        a[++p].x=now,a[p].y=val+w[now];
    }
}
ll dfs1(ll now){
    if(visit[now]) return 0;
    visit[now]=1;
    return dfs1(fa[now])+w[now];
}
int main(){
    int t;
    cin>>t;
    int id=0;
    while(t--){
        init();
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) fa[i]=i;
        for(int i=1;i<=n;i++) scanf("%d",&w[i]);
        for(int i=1;i<=n-1;i++){
            int u,v;
            scanf("%d%d",&u,&v);
            add(u,v);
            fa[v]=u;
        }
        dfs(1,0);
        sort(a+1,a+p+1,cmp);
        for(int i=1;i<=n;i++){
            ans[i]=dfs1(a[i].x);
        }
        sort(ans+1,ans+p+1,greater<ll>());
        ll res=0;
        for(int i=1;i<=m;i++){
            res+=ans[i];
        }
        printf("Case #%d: %lld\n",++id,res);
    }
    //system("pause");
}