从1点出发,走k步,最多能够拿到多少苹果?
同样考虑是否重复走边。
需要注意的是,这一道题没有要全部遍历的限制,也没有多个人一起走
讨论点进去了要不要出来。这个作为状态。
然后对于某个儿子,讨论进去了要不要出来、出来了回去还是继续下一个儿子。以此转移。
#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<ctime>
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;
}