zoj 3201 简单树形dp Tree of Tree

#include<iostream>
#include<cstdio>
/// 从 一棵树上选取k 个节点的子树的节点和的最大值
/// 状态转移方程 ： dp[u][j]=max(dp[u][tt]+dp[v][j-tt],dp[u][j]);
/// 表示从父节点选取tt个，从孩子节点选取j-tt个的最大值
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
int n,k,dp[102][102],sum[102];
vector<int>p[102];
void dfs(int u,int fa)
{
    int i,j,tt, v;
    dp[u][1]=sum[u];
    for(i=0; i<p[u].size(); i++)
    {
        v=p[u][i];
        if(v==fa)continue;
        dfs(v,u);
        for(j=k; j>=1; j--)
        {
            for(tt=1; tt<=j; tt++)
            {
                dp[u][j]=max(dp[u][tt]+dp[v][j-tt],dp[u][j]);
            }
        }
    }
}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        int i,j,u,v,ans=0;
        for(i=0; i<=n; i++)
        {
            p[i].clear();
            for(j=0; j<=k; j++)
                dp[i][j]=0;
        }

        for(i=1; i<=n; i++)cin>>sum[i];
        for(i=1; i<n; i++)
        {
            cin>>u>>v;
            u++,v++;
            p[u].push_back(v);
            p[v].push_back(u);
        }
        dfs(1,-1);
        for(i=1; i<=n; i++)
        {
            ans=max(ans,dp[i][k]);
        }
        cout<<ans<<endl;
    }
    return 0;
}