设 d p [ u ] [ x ] dp[u][x] dp[u][x] 表示以 u u u 为根的子树,选 x x x 个黑点时子树内的边对答案的最大贡献。
这个状态设得比较巧妙,求的是子树内边的贡献和。
d p [ u ] [ x ] = d p [ u ] [ x ] + d p [ v ] [ y ] + e d g e ( x , y ) ∗ ( y ∗ ( K − y ) + ( s i z [ v ] − y ) ∗ ( n − K − s i z [ v ] + y ) ) dp[u][x]=dp[u][x]+dp[v][y]+\\edge(x,y)*(y*(K-y)+(siz[v]-y)*(n-K-siz[v]+y)) dp[u][x]=dp[u][x]+dp[v][y]+edge(x,y)∗(y∗(K−y)+(siz[v]−y)∗(n−K−siz[v]+y))
时间复杂度 O ( n 2 ) O(n^2) O(n2) 。可以算是树上背包的变形。
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define PII pair<int,int>
#define ll long long
#define f(x) 1ll*x*(x-1)/2
#define All(a) a.begin(),a.end()
using namespace std;
const int mx = 2005;
int n, K, siz[mx];
ll dp[mx][mx];
vector<PII> g[mx];
void dfs(int u, int fa) {
siz[u] = 1;
for (auto v : g[u]) {
if (v.first == fa)
continue;
dfs(v.first, u);
for (int i = siz[u]; i >= 0; i--) {
for (int j = siz[v.first]; j >= 0; j--) {
dp[u][i + j] = max(dp[u][i + j], dp[u][i] + dp[v.first][j] + 1ll * v.second * (j * (K - j) +
(siz[v.first] - j) * (n - K - siz[v.first] + j)));
}
}
siz[u] += siz[v.first];
}
}
signed main() {
// freopen("data.in","r",stdin);
scanf("%d%d", &n, &K);
for (int i = 1; i < n; i++) {
int u, v, w;
scanf("%d%d%d", &u, &v, &w);
g[u].push_back(make_pair(v, w));
g[v].push_back(make_pair(u, w));
}
dfs(1, 0);
printf("%lld", dp[1][K]);
}