【树形DP】17.2.6 T4 苹果树 题解

题目描述：
jyb 有一天梦见自己变成了一只萌萌的苹果虫。
jyb 作为一只虫的时间只有t 小时，之后他就去作茧化蝶，在这t 小时内，他需要在苹果树上尽可能多地吃苹果。
苹果树可以看做一棵有n 个节点n-1 条边的树，每个节点有ai 个苹果，jyb每爬过一条边需要1 小时的时间，吃苹果时间忽略不计。jyb 在每一个小时内可以原地休息，也可以在一条树枝上来回爬。
现在jyb 从1 号点开始爬，请问jyb 最多能吃到多少个苹果？
【输入格式】
第一行两个正整数n, t。表示苹果树节点个数和jyb 的时间。
第二行n 个数，表示第i 个点的苹果数ai。
接下来n-1 行，每行两个数u, v，表示u, v 是树上一条边。
【输出格式】
一个整数，jyb 最多能吃到的苹果数目。
【输入样例1】
2 1
0 11
1 2
【输出样例1】
11
【输入样例2】
3 2
0 1 2
1 2
1 3
5
【输出样例2】
2
【数据范围】
对于20%的数据，n <= 10, t <= 5
对于100%的数据，0 <=ai <= 1000, 1<=n<=100, 0<=t<=200

关键是转移啊……
转移的时候，如果当前要走回来，则所有子节点都要走回来。如果当前节点不走回来，则除一个节点外其他的都要走回来。可以据此列出以下几个方程：
dp[u][j][0] = max(dp[u][j][0], dp[u][j - k][0] + dp[v][k - 2][0]);
dp[u][j][1] = max(dp[u][j][1], dp[u][j - k][1] + dp[v][k - 2][0]);
dp[u][j][1] = max(dp[u][j][1], dp[u][j - k][0] + max(dp[v][k - 1][1],dp[v][k - 1][0]));

附STD：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <set>
#include <queue>
#include <algorithm>
#include <vector>
#include <cstdlib>
#include <cmath>
#include <ctime>
#include <stack>
#define INF 2100000000
#define ll long long
#define clr(x)  memset(x,0,sizeof(x))

using namespace std;

const int maxn = 105;
vector<int> e[maxn];
int dp[maxn][205][2];
int n,m,a[maxn];

void dfs(int u, int p)
{
    int len = e[u].size();
    for(int tv = 0; tv <= len-1; tv++)
    {
        int v = e[u][tv];
        if(v == p) continue;
        dfs(v, u);
        for(int j = m; j >= 1; j--)
        {
            dp[u][j][0] = max(dp[u][j][0], dp[u][j-1][0]), dp[u][j][1] = max(dp[u][j][1], dp[u][j-1][1]);
            for(int k = 1; k <= j; k++)
            {
                if(k >= 2)
                    dp[u][j][0] = max(dp[u][j][0], dp[u][j-k][0] + dp[v][k-2][0]),
                    dp[u][j][1] = max(dp[u][j][1], dp[u][j-k][1] + dp[v][k-2][0]);
                if(k >= 1) dp[u][j][1] = max(dp[u][j][1], dp[u][j-k][0]+max(dp[v][k-1][1], dp[v][k-1][0]));
            }
        }
    }
}

int main() {
    freopen("apple.in","r",stdin);
    freopen("apple.out","w",stdout);
    while(scanf("%d%d",&n,&m) != EOF)
    {
        int u,v;
        clr(dp);
        for(int i = 1; i <= n; i++) e[i].clear();
        for(int i = 1; i <= n; i++)
        {
            scanf("%d",&a[i]), dp[i][0][0] = a[i];
            for(int j = 1; j <= m; j++) dp[i][j][0] = dp[i][j][1] = a[i];
        }
        for(int i = 1; i <= n-1; i++) scanf("%d%d",&u,&v), e[v].push_back(u), e[u].push_back(v);
        dfs(1, 0);
        printf("%d\n",max(dp[1][m][0], dp[1][m][1]));
    }
    return 0;
}