[JZOJ 5788] 餐馆

思路：

考虑树形dp。

我们设\(dp[i][j][0/1]\)表示在\(i\)为根的子树中花费\(j\)单位时间，最终回到/不必回到\(i\)的最大收益。

转移三种：

\(dp[x][j][0] = max(dp[x][j][0],dp[x][j - k - 2][0] + dp[y][k][1]);\)

\(dp[x][j][0] = max(dp[x][j][0],dp[x][j - k - 1][1] + dp[y][k][0]);\)

\(dp[x][j][1] = max(dp[x][j][1],dp[x][j - k - 2][1] + dp[y][k][1]);\)

搞定.

#include <bits/stdc++.h>
using namespace std;
inline int read() {
	int q=0,f=1;char ch = getchar();
	while(!isdigit(ch)){
		if(ch=='-')f = -1;ch=getchar();
	}
	while(isdigit(ch)){
		q=q*10+ch-'0';ch=getchar();
	}
	return q*f;
}
const int maxn = 600;
struct edge {
	int to;
	int nxt;
}e[maxn << 1];
int cnt;
int head[maxn<<1];
inline void add_edge(int u,int v) {
	e[++cnt].to = v;
	e[cnt].nxt = head[u];
	head[u] = cnt;
}
inline void Add_edge(int u,int v) {
	add_edge(u,v);
	add_edge(v,u);
	return;
}
int dp[maxn][maxn][2];
int m;int n;
int a[maxn];
inline void dfs(int x,int fa) {
	for(int i = head[x];i;i=e[i].nxt) {
		int y = e[i].to;
		if(y == fa) continue;
		dfs(y,x);
		for(int j = m;j >= 2; --j) {
			for(int k = 0;k <= m; ++k) {
				if(j - k - 2 < 0) break;
				dp[x][j][0] = max(dp[x][j][0],dp[x][j - k - 2][0] + dp[y][k][1]);
			}
		}
		for(int j = m;j >= 1; --j) {
			for(int k = 0;k <= m; ++k) {
				if(j - k - 1 < 0) break;
				dp[x][j][0] = max(dp[x][j][0],dp[x][j - k - 1][1] + dp[y][k][0]);
			}
		}
		for(int j = m;j >= 2; --j) {
			for(int k = 0;k <= m; ++k) {
				if(j - k - 2 < 0) break;
				dp[x][j][1] = max(dp[x][j][1],dp[x][j - k - 2][1] + dp[y][k][1]);
			}
		}
	}
	for(int i = m;i >= 1; --i) {
		dp[x][i][0] = max(dp[x][i][0],dp[x][i - 1][0] + a[x]);
		dp[x][i][1] = max(dp[x][i][1],dp[x][i - 1][1] + a[x]);
	}
}
int main () {
	#ifdef ONLINE_JUDGE
		freopen("dostavljac.in","r",stdin);
		freopen("dostavljac.out","w",stdout);
	#endif
	n = read(),m = read();
	for(int i = 1;i <= n; ++i) {
		a[i] = read();
	}
	for(int i = 1;i < n; ++i) {
		int x = read(),y = read();
		Add_edge(x,y);
	}
	dfs(1,0);
	printf("%d\n",dp[1][m][0]);
	return 0;
}