Tree of Tree ZOJ - 3201 树形dp

题目链接：https://vjudge.net/problem/ZOJ-3201
题意：一棵树，求满足有k个节点的子树的最大节点权值和。输入保证合法。
思路：dp[i][j]表示以i为根，包含j个节点时的答案。注意更新答案时的枚举顺序。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
using namespace std;
int w[105],n,k,ans,dp[105][105];
vector <int> g[105];
void dfs(int cur,int pre)
{
    dp[cur][1]=w[cur];
    for(int i=0; i<g[cur].size(); i++)
    {
        if(g[cur][i]==pre)
            continue;
        dfs(g[cur][i],cur);
        for(int j=k; j>=1; j--)//注意此处的枚举顺序 这样才能保证cur
        {//的子节点状态只用一次，不会有重复
            for(int k=0; k<j; k++)//表示有几个节点分配给cur的子节点
            {
                dp[cur][j]=max(dp[cur][j],dp[cur][j-k]+dp[g[cur][i]][k]);
            }
        }
    }
    ans=max(dp[cur][k],ans);
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        ans=0;
        memset(dp,0,sizeof(dp));
        for(int i=0; i<=n; i++)
        {
            g[i].clear();
        }
        for(int i=0; i<n; i++)
        {
            scanf("%d",&w[i]);
        }
        for(int i=0; i<n-1; i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            g[a].push_back(b);
            g[b].push_back(a);
        }
        dfs(0,-1);
        printf("%d\n",ans);
    }
    return 0;
}