【树形dp】PKU-2486-Apple Tree

在原来很裸的树形dp上加了些限制条件，不过知道状态转移还是很好做的，但状态转移不好想啊，都是看了人家的结题报告后才知道的！

题目

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#include<stack>
using namespace std;
#define FRE freopen("a.txt","r",stdin);
#define inf 999999999
#define N 105
vector<int> v[N];
int m,dp[2][N][2*N],vis[N];
void dfs(int now)
{
    for(int i=0;i<=m;i++)dp[0][now][i]=dp[1][now][i]=dp[0][now][0];             //在以now为跟的子树里，不管怎样走，其值至少为now的权值
    vis[now]=1;
    for(int i=0;i<v[now].size();i++)
    {
        int u=v[now][i];
        if(vis[u])continue;
        dfs(u);
        for(int j=m-1;j>=0;j--)
        {
            for(int k=m-j;k>=0;k--)
            {
                dp[0][now][j+k+2]=max(dp[0][now][j+k+2],dp[0][now][j]+dp[0][u][k]);             //在以u为跟的子树走了一圈回到now点，然后在除开u的子树再走一圈再回到now
                dp[1][now][j+k+1]=max(dp[1][now][j+k+1],dp[0][now][j]+dp[1][u][k]);             //在除开u的子树走一圈回到now点，然后在以u为跟的子树再走了一圈但不回到now
                dp[1][now][j+k+2]=max(dp[1][now][j+k+2],dp[1][now][j]+dp[0][u][k]);             //在以u为跟的子树走了一圈回到now点，然后在除开u的子树在走一圈但不回到now
            }
        }
    }
}
int main()
{
    //FRE;
    int n;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&dp[0][i][0]);
            v[i].clear();
        }
        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);
        }
        memset(vis,0,sizeof(vis));
        dfs(1);
        printf("%d\n",dp[1][1][m]);
    }
    return 0;
}