社区讨论
8pts WA玄关
P2015二叉苹果树参与者 2已保存回复 3
讨论操作
快速查看讨论及其快照的属性,并进行相关操作。
- 当前回复
- 3 条
- 当前快照
- 1 份
- 快照标识符
- @m091ptyl
- 此快照首次捕获于
- 2024/08/25 12:04 去年
- 此快照最后确认于
- 2025/11/04 22:29 4 个月前
CPP
#include<bits/stdc++.h>
#define son first
#define val second
using namespace std;
const int N = 110;
vector <pair <int, int> > e[N];
int n, k, dp[N][N], value[N], dep[N], s[N][2], siz[N];
void dfs(int x, int fa)
{
if (x != 0 && e[x].size() == 1)
{
dp[x][0] = value[x];
return;
}
if (s[x][0])
dfs(s[x][0], x);
if (s[x][1])
dfs(s[x][1], x);
if (s[x][0] && s[x][1])
{
for (int i = 0; i <= k; i++)
for (int j = 0; j <= k - i; j++)
dp[x][i + j] = max(dp[s[x][0]][i] + dp[s[x][1]][j], dp[x][i + j]);
for (int i = 0; i <= k - siz[s[x][0]]; i++)
dp[x][i + siz[s[x][0]]] = max(dp[s[x][1]][i], dp[x][i + siz[s[x][0]]]);
for (int i = 0; i <= k - siz[s[x][1]]; i++)
dp[x][i + siz[s[x][1]]] = max(dp[s[x][0]][i], dp[x][i + siz[s[x][1]]]);
dp[x][min(k, siz[x] - 1)] = max(dp[x][min(k, siz[x] - 1)], 0);
}
else if (s[x][0])
{
for (int i = 0; i <= k; i++)
dp[x][i] = max(dp[x][i], dp[s[x][0]][i]);
dp[x][min(k, siz[x] - 1)] = max(dp[x][min(k, siz[x] - 1)], 0);
}
for (int i = 0; i <= k; i++)
dp[x][i] += value[x];
return;
}
void build(int x, int fa)
{
dep[x] = dep[fa] + 1;
siz[x] = 1;
if (x != 1 && e[x].size() == 1)
return;
int p = 0;
for (auto i : e[x])
{
if (i.son == fa)
continue;
s[x][p++] = i.son;
value[i.son] = i.val;
build(i.son, x);
siz[x] += siz[i.son];
}
return;
}
int main()
{
memset(dp, 0xc0, sizeof dp);
cin >> n >> k;
for (int i = 1, u, v, w; i < n; i++)
{
scanf("%d%d%d", &u, &v, &w);
e[u].push_back({v, w});
e[v].push_back({u, w});
}
build(1, 0);
dfs(1, 0);
printf("%d", dp[1][k]);
return 0;
}
回复
共 3 条回复,欢迎继续交流。
正在加载回复...