【HIHOCODER 1055】刷油漆（树上背包）

描述

小Ho有着一棵灰常好玩的树玩具！这棵树玩具是由N个小球和N-1根木棍拼凑而成，这N个小球都被小Ho标上了不同的数字，并且这些数字都是处于1..N的范围之内，每根木棍都连接着两个不同的小球，并且保证任意两个小球间都不存在两条不同的路径可以互相到达。没错，这次说的还是这棵树玩具的故事！

小Ho的树玩具的质量似乎不是很好，短短玩了几个星期，便掉漆了！

“简直是一场噩梦！”小Ho拿着树玩具眼含热泪道。

“这有什么好忧伤的，自己买点油漆刷一刷不就行了？”小Hi表示不能理解。

“还可以这样？”小Ho顿时兴高采烈了起来，立马跑出去买回来了油漆，但是小Ho身上的钱却不够——于是他只买回了有限的油漆，这些油漆最多能给M个结点涂上颜色，这就意味着小Ho不能够将他心爱的树玩具中的每一个结点都涂上油漆！

小Ho低头思索了半天——他既不想只选一部分结点补漆，也不想找小Hi借钱，但是很快，他想出了一个非常棒的主意：将包含1号结点的一部分连通的结点进行涂漆（这里的连通指的是这一些涂漆的结点可以互相到达并且不会经过没有涂漆的结点），然后将剩下的结点拆掉！

那么究竟选择哪些结点进行涂漆呢？小Ho想了想给每个结点都评上了分——他希望最后留下来，也就是涂漆了的那些结点的评分之和可以尽可能的高！

那么，小Ho该如何做呢？

输入

每个测试点（输入文件）有且仅有一组测试数据。

每组测试数据的第一行为两个整数N、M，意义如前文所述。

每组测试数据的第二行为N个整数，其中第i个整数Vi表示标号为i的结点的评分

每组测试数据的第3~N+1行，每行分别描述一根木棍，其中第i+1行为两个整数Ai，Bi，表示第i根木棍连接的两个小球的编号。

对于100%的数据，满足N<=10^2，1<=Ai<=N, 1<=Bi<=N, 1<=Vi<=10^3, 1<=M<=N

输出

第1行：1个实数d，保留3位小数(四舍五入)对于每组测试数据，输出一个整数Ans，表示使得涂漆结点的评分之和最高可能是多少。

样例输入

10 4
370 328 750 930 604 732 159 167 945 210
1 2
2 3
1 4
1 5
4 6
4 7
4 8
6 9
5 10

样例输出

题解

设f[rt][j]表示以节点rt为根的子树在油漆使用j个的情况下所能得到的最大分数，那么它的每个子节点获取分数及其所需都可以看成一个物品，类01背包

For i=each child of rt
For j=m .. 2
     For k=1 .. j-1
           f[rt][j]=max(f[rt][j],f[rt][j-k]+f[i][k])

#include <bits/stdc++.h>
#define ll long long
#define inf 1000000000
#define PI acos(-1)
#define bug puts("here")
#define REP(i,x,n) for(int i=x;i<=n;i++)
#define DEP(i,n,x) for(int i=n;i>=x;i--)
#define mem(a,x) memset(a,x,sizeof(a))
using namespace std;
inline int read(){
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int N=205;
ll f[N][N];
struct node{
    int to,nxt;
}Path[N<<1];
int tot=0,head[N],n,m;
void Addedge(int u,int v){
     Path[++tot]=node{v,head[u]};
     head[u]=tot;
}
void dfs(int cur,int fa){
     for(int i=head[cur];i;i=Path[i].nxt){
        if(Path[i].to==fa) continue;
        dfs(Path[i].to,cur);
        DEP(j,m,2){
            REP(k,1,j-1){
                f[cur][j]=max(f[cur][j],f[cur][j-k]+f[Path[i].to][k]);
            }
        }
     }
}
int main(){
   // while(1)
    {
        n=read();m=read();
        REP(i,1,n) f[i][1]=read();
        mem(head,0);tot=0;
        REP(i,1,n-1){
             int u=read(),v=read();
             Addedge(u,v);
             Addedge(v,u);
        }
        dfs(1,0);
        printf("%lld\n",f[1][m]);
    }
    return 0;
}