【香蕉OI】商贸往来【HAOI 2015】 树上染色（树型 DP）

题意

一棵 $n$ 个点的树，选 $m$ 个黑点，其余白点，最大化两两黑点之间距离和两两白点之间距离之和。

思路

DP 很好想， $dp[i][j]$ 表示 $i$ 为根的子树中选了 $j$ 个黑点的情况下，这棵子树中的黑点和白点对答案的贡献。

因为子树内部有 $j$ 个黑点，那么外部就有 $m-j$ 个黑点，那么内部黑点的贡献就是内部两两距离和加上所有黑点到 $i$ 的距离乘上外部黑点个数。很好转移。

主要是如何证明这个看似 $O(n^3)$ 的 DP 是 $O(n^2)$ 的。这里摘抄一下 扶苏的博客 。

发现对于每个点，做的工作相当于先把 $i$ 看做一个点，然后枚举每个儿子，做 $O(siz[i]\ast siz[son])$ 的工作，把这个儿子接到 $i$ 上。

而这个复杂度可以看作枚举 $son$ 中的一个点和之前的 $i$ 的子树中的一个点。发现每一个点对只可能在他们的 LCA 处枚举到，所以是 $O(n^2)$ 的。

你以为你打的是暴力？？？

代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
typedef long long LL;
const int N = 5000 + 10, M = N<<1;
int n, k, siz[N];
int h[N], ecnt, v[M], w[M], nxt[M];
LL dp[N][N];
 
template<class T>inline void read(T &x){
    x = 0; bool fl = 0; char c = getchar();
    while (!isdigit(c)){if (c == '-') fl = 1; c = getchar();}
    while (isdigit(c)){x = (x<<3)+(x<<1)+c-'0'; c = getchar();}
    if (fl) x = -x;
}
 
void add(int x, int y, int z){
    v[++ecnt] = y; w[ecnt] = z; nxt[ecnt] = h[x];
    h[x] = ecnt;
}
 
namespace Solver1
{
    void dfs(int u, int fa)
    {
        siz[u] = 1;
        for (int i = h[u]; i; i = nxt[i])
            if (v[i] != fa){
                dfs(v[i], u);
                for (int o = min(k, siz[u]+siz[v[i]]); o >= 0; -- o)
                    for (int p = max(o-siz[u], 0LL), qw = min(siz[v[i]], o); p <= qw; ++ p){
                        dp[u][o] = max(dp[u][o], dp[u][o-p] + dp[v[i]][p] + w[i]*(1LL*(n-siz[v[i]])*siz[v[i]]+2LL*(k-p)*p-1LL*(n-siz[v[i]])*p-1LL*siz[v[i]]*(k-p)));
                    }
                siz[u] += siz[v[i]];
            }
    }
}

signed main()
{
    read(n); read(k);
    for (int i = 1; i < n; ++ i){
        int x, y, z;
        read(x); read(y); read(z);
        add(x, y, z); add(y, x, z);
    }
    Solver1::dfs(1, 0);
    printf("%lld\n", dp[1][k]);
    return 0;
}