记录编号 602704 评测结果 AAAAAAAAAA
题目名称 4085.外卖 最终得分 100
用户昵称 GravatarOTTF 是否通过 通过
代码语言 C++ 运行时间 0.362 s
提交时间 2025-07-05 15:22:22 内存使用 4.47 MiB
显示代码纯文本

#include <algorithm>
#include <cstdio>
#include <iostream>
#include <vector>

using namespace std;

constexpr int N = 514;

int n;
int m;
long long nums[N];
vector<int> tree[N];
int dp[N][N][2];

void dp1 (int now, int dad) {
	dp[now][1][0] = dp[now][1][1] = nums[now];
	for (auto son : tree[now]) {
		if (son == dad) {
			continue;
		}

		dp1 (son, now);

		for (int i = m; i >= 1; i--) {
			for (int j = 1; j <= i; j++) {
				dp[now][i][0] = max (dp[now][i][0], dp[now][i - j][1] + dp[son][j - 1][0]);
				if (j >= 2) {
					dp[now][i][0] = max (dp[now][i][0], dp[son][j - 2][1] + dp[now][i - j][0]);
					dp[now][i][1] = max (dp[now][i][1], dp[son][j - 2][1] + dp[now][i - j][1]);
				}
			}
		}
	}
}

int main () {
	
	freopen ("food.in", "r", stdin);
	freopen ("food.out", "w", stdout);

	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		cin >> nums[i];
	}
	int u, v;
	for (int i = 1; i < n; i++) {
		cin >> u >> v;
		tree[u].push_back(v);
		tree[v].push_back(u);
	}

	dp1 (1, 0);

	cout << max (dp[1][m][0], dp[1][m][1]) << endl;
	
	return 0;
}