[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;
}