POJ 2486

【题目分析】
树型DP

---

【代码】

#include <cstdio>
#include <cmath>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
vector <int> v[101];
int dp[101][201][2],w[101];
int n,m;
inline void dfs(int k,int fa)
{
    for (int i=0;i<v[k].size();++i)
    if (v[k][i]!=fa)
    {
        dfs(v[k][i],k);
        for (int j=m;j>=1;--j)
            for (int t=1;t<=j;++t)
            {
                dp[k][j][0]=max(dp[k][j][0],dp[k][j-t][1]+dp[v[k][i]][t-1][0]);
                dp[k][j][0]=max(dp[k][j][0],dp[k][j-t][0]+dp[v[k][i]][t-2][1]);
                dp[k][j][1]=max(dp[k][j][1],dp[k][j-t][1]+dp[v[k][i]][t-2][1]);
            }
    }
}
int main()
{
    while (~scanf("%d%d",&n,&m))
    {
        memset(dp,0,sizeof dp);
        for (int i=1;i<=n;++i)
        {
            v[i].clear();
            scanf("%d",&w[i]);
            for (int j=0;j<=m;++j)
                dp[i][j][0]=dp[i][j][1]=w[i];
        }
        for (int i=1;i<n;++i)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            v[a].push_back(b);
            v[b].push_back(a);
        }
        dfs(1,0);
        printf("%d\n",dp[1][m][0]);
    }
}