[POJ2486] Apple Tree [树形dp]

[$\mathfrak{L}\mathfrak{i}\mathfrak{n}\mathfrak{k}$]

---

从1点出发，走k步，最多能够拿到多少苹果？
同样考虑是否重复走边。
需要注意的是，这一道题没有要全部遍历的限制，也没有多个人一起走

讨论点进去了要不要出来。这个作为状态。
然后对于某个儿子，讨论进去了要不要出来、出来了回去还是继续下一个儿子。以此转移。

---

#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define add_edge(a, b) nxt[++tot]=head[a],head[a]=tot,to[tot]=b
int head[105], nxt[205], to[205];
int n, k, tot;
int w[105];
int f[105][205][2];
void dfs(int x, int fa) {
	for (int i = 0; i <= k; ++i) f[x][i][0] = f[x][i][1] = w[x];
	for (int i = head[x]; i; i = nxt[i]) {
		if (to[i] == fa) continue;
		dfs(to[i], x);
		for (int j = k; j >= 0; --j) {
			for (int p = 0; p <= j; ++p) {
				if (j >= p + 2) f[x][j][0] = max(f[x][j][0], f[to[i]][p][0] + f[x][j-p-2][0]);
				if (j >= p + 1) f[x][j][1] = max(f[x][j][1], f[to[i]][p][1] + f[x][j-p-1][0]);
				if (j >= p + 2) f[x][j][1] = max(f[x][j][1], f[to[i]][p][0] + f[x][j-p-2][1]);
			}
		}
	}
}
int main() {
	while (~scanf("%d%d", &n, &k)) {
		tot = 0;
		memset(head, 0, sizeof(head));
		memset(f, 0, sizeof(f));
		for (int i = 1; i <= n; ++i) {
			scanf("%d", &w[i]);
		}
		for (int a, b, i = 1; i < n; ++i) {
			scanf("%d%d", &a, &b);
			add_edge(a, b);
			add_edge(b, a);
		}
		dfs(1, 0);
		printf("%d\n", max(f[1][k][0], f[1][k][1]));
	}
	return 0; 
}