【JZOJ5788】餐馆

Description

 K妹的胡椒粉大卖，这辣味让食客们感到刺激，许多餐馆也买这位K妹的账。有N家餐馆，有N-1条道路，这N家餐馆能相互到达。K妹从1号餐馆开始。每一个单位时间，K妹可以在所在餐馆卖完尽量多的胡椒粉，或者移动到有道路直接相连的隔壁餐馆。第i家餐馆最多需要A[i]瓶胡椒粉。K妹有M个单位的时间，问她最多能卖多少胡椒粉。

Input

第一行有两个正整数N，M。
第二行描述餐馆对胡椒粉的最大需求量，有N个正整数，表示A[i]。
接下来有N-1行描述道路的情况，每行两个正整数u，v，描述这条道路连接的两个餐馆。

Output

一个整数，表示她最多能卖的胡椒粉瓶数。

Sample Input

样例1输入
3 5
9 2 5
1 2
1 3

样例2输入
4 5
1 1 1 2
1 2
2 3
3 4

样例3输入
5 10
1 3 5 2 4
5 2
3 1
2 3
4 2
 

Sample Output

样例1输出
14

样例2输出
3

样例3输出
15
 

Data Constraint

对于10%的数据，N≤20。
对于50%的数据，N≤110。
对于100%的数据1 ≤ N, M ≤ 500，1 ≤ A[i]≤ 10^6，
第5到第10个测试点都有多个子测试。

Hint

在样例1的中，辣妹到达城市2后就恰好没时间卖辣椒粉了。

题解：

就算知道了是树形DP也不可能想出转移的，这辈子都不可能的= =
看了一下题解，大致明白了思路，这里复制一下题解内容
设f[i][j][0/1]
f[i][j][0]表示以 i 为根的子树中，花费 j 单位时间，最终回到 i 的最大收益。
f[i][j][1]表示以 i 为根的子树中，花费 j 单位时间，最终不必回到 i 的最大收益。
先走之前的子树回到根，再走下一颗子树不回根：
dp[i][j][0] + dp[son][k][1] --> dp[i][j+k+1][1]
先走之前的子树回到根，再走下一颗子树也回根：
dp[i][j][0] + dp[son][k][0] --> dp[i][j+k+2][0]
先下一颗子树回到根，再走之前的子树不回根：
dp[i][j][1] + dp[son][k][0] --> dp[i][j+k+2][1]
枚举下一个子树后，倒序枚举j和k
整个过程转移是从叶节点向上推进，因此在枚举子树的下面要初始化一下
时间复杂度为O(n^2)，具体详见代码= = 

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<cctype>
#include<cstdlib>
#include<climits>
#include<iomanip>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#define MAXA 505
#define ipt(x) scanf("%d",&x)
using namespace std;
typedef long long LL;

int n,m,Ans;
int Pepper[MAXA],f[MAXA][MAXA][2];
bool vis[MAXA];
vector<int> son[MAXA];
void DP(int x) {
	vis[x] = 1;
	for(int i=0;i<son[x].size();i++) {
		int y = son[x][i];
		if(!vis[y]) {
			DP(y);
			for(int j=m;j>=0;j--)
		        for(int k=m-j;k>=0;k--) {
		    	if(j + k + 1 <= m)
				   f[x][j+k+1][1] = max(f[x][j+k+1][1],f[x][j][0] + f[y][k][1]);
				if(j + k + 2 <= m)
				   f[x][j+k+2][0] = max(f[x][j+k+2][0],f[x][j][0] + f[y][k][0]);
				if(j + k + 2 <= m)
				   f[x][j+k+2][1] = max(f[x][j+k+2][1],f[x][j][1] + f[y][k][0]); 
			}
		}
		
	}
	for(int i=m;i>=1;i--) {
		f[x][i][0] = max(f[x][i][0],f[x][i-1][0] + Pepper[x]);
		f[x][i][1] = max(f[x][i][1],f[x][i-1][1] + Pepper[x]);
	}
}
int main() {
	freopen("dostavljac.in","r",stdin);
	freopen("dostavljac.out","w",stdout);
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
	    scanf("%d",&Pepper[i]);
	for(int i=1;i<n;i++) {
		int u,v;
		scanf("%d %d",&u,&v);
		son[u].push_back(v);
		son[v].push_back(u);
	}
	DP(1);
	printf("%d",max(f[1][m][1],f[1][m][0]));
}