【POJ2486】树形DP

AC代码见下

//POJ2486树形DP
//这个应该算是一个比较简单的板子题
//根给了，信息也给了，只是有一个步数限制，以前DP中的限制
//状态转移方程
#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
int h[105];
vector<int> v[105];
int n,k;
int dp[105][205][2];   //建立状态转移方程
void dfs(int x,int fa){
    //if(v[x].size()==0) return;
    for(int i=0;i<v[x].size();i++){
        int g=v[x][i];
        if(g==fa) continue;
        dfs(g,x);
    for(int j=k;j>=1;j--){
        for(int t=1;t<=j;t++)
			{
				dp[x][j][0]=max(dp[x][j][0],dp[x][j-t][0]+dp[g][t-2][0]);
				dp[x][j][1]=max(dp[x][j][1],dp[x][j-t][0]+dp[g][t-1][1]);
				dp[x][j][1]=max(dp[x][j][1],dp[x][j-t][1]+dp[g][t-2][0]);
			}
        }
    }
}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF){
        for(int i=0;i<=n;i++)
            v[i].clear();                 //记得清空数组
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)
            {
                scanf("%d",&h[i]);
                for(int j=0;j<=k;j++)
                    dp[i][j][0]=dp[i][j][1]=h[i];
            }
        for(int i=1;i<n;i++){
            int x,y;
            scanf("%d%d",&x,&y);
            v[x].push_back(y);
            v[y].push_back(x);
        }
        dfs(1,-1);
        printf("%d\n",max(dp[1][k][0],dp[1][k][1]));
    }
    return 0;
}