zoj3201

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3201

#include<bits/stdc++.h>
using namespace std;
int vis[105];
int val[105];
vector<int> v[105];
int f[105][105];
int n,k;
int ans=0;
void dfs(int x)
{
    f[x][1]=val[x];
    vis[x]=1;
    for(int i=0; i<v[x].size(); i++)
    {
        int y=v[x][i];
        if(vis[y]) continue;
//        printf("??");
        dfs(y);
        for(int j=k; j>=2; j--)
            for(int k1=1; k1<j; k1++)
            {
                f[x][j]=max(f[x][j],f[x][j-k1]+f[y][k1]);
//                printf("??");
            }
    }
    ans=max(ans,f[x][k]);
}
int main()
{

    while(~scanf("%d%d",&n,&k))
    {
        for(int i=0; i<n; i++)
        {
            scanf("%d",&val[i]);
            v[i].clear();
        }
        memset(f,0,sizeof(f));
        memset(vis,0,sizeof(vis));
        for(int i=1; i<=n-1; i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            v[a].push_back(b);
            v[b].push_back(a);
        }
        ans=0;
        dfs(0);
        printf("%d\n",ans);
    }

}