[BZOJ4033][HAOI2015]树上染色

4033: [HAOI2015]树上染色

Time Limit: 10 Sec Memory Limit: 256 MB
Submit: 2108 Solved: 901
[Submit][Status][Discuss]

Description

有一棵点数为N的树，树边有边权。给你一个在0~N之内的正整数K，你要在这棵树中选择K个点，将其染成黑色，并

将其他的N-K个点染成白色。将所有点染色后，你会获得黑点两两之间的距离加上白点两两之间距离的和的收益。

问收益最大值是多少。

Input

第一行两个整数N,K。

接下来N-1行每行三个正整数fr,to,dis，表示该树中存在一条长度为dis的边(fr,to)。

输入保证所有点之间是联通的。

N<=2000,0<=K<=N

Output

输出一个正整数，表示收益的最大值。

Sample Input

5 2
1 2 3
1 5 1
2 3 1
2 4 2

Sample Output

17
【样例解释】
将点1,2染黑就能获得最大收益。

设$dp[u][i]$表示在以$u$为根的子树中有$i$个黑点时每条边对答案的贡献和最大值

每次把子树的答案合并，只需考虑连接孩子的这一条边的两边各有多少黑点白点即可

每次转移相当于枚举点对

所以时间复杂度为$O\left(n^2\right)$

#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn = 2000 + 10;
int n, k;
struct Edge{int to, val, next;Edge(){}Edge(int _t, int _v, int _n): to(_t), val(_v), next(_n){}
}e[maxn * 2];
int fir[maxn] = {0}, cnt = 0;
inline void ins(int u, int v, int w){e[++cnt] = Edge(v, w, fir[u]); fir[u] = cnt;e[++cnt] = Edge(u, w, fir[v]); fir[v] = cnt;
}
int siz[maxn];
ll f[maxn][maxn] = {0}, g[maxn];
void dfs(int u, int fa){siz[u] = 1;f[u][0] = f[u][1] = 0;for(int v, i = fir[u]; i; i = e[i].next){v = e[i].to;if(v == fa) continue;dfs(v, u);for(int x = 0; x <= siz[u] + siz[v]; x++) g[x] = 0;for(int x = 0; x <= siz[u]; x++)for(int y = 0; y <= siz[v]; y++)g[x + y] = max(g[x + y], f[u][x] + f[v][y] + (ll)e[i].val * (y * (k - y) + (siz[v] - y) * (n - k + y - siz[v])));siz[u] += siz[v];for(int x = 0; x <= siz[u]; x++)f[u][x] = max(f[u][x], g[x]);}
}
int main(){scanf("%d %d", &n, &k);for(int u, v, w, i = 1; i < n; i++){scanf("%d %d %d", &u, &v, &w);ins(u, v, w);}dfs(1, 0);printf("%lld\n", f[1][k]);return 0;
}

转载于:https://www.cnblogs.com/ruoruoruo/p/7669063.html