重刷树形背包计划!

一.洛谷P3177

题目大意:

给你一颗含n个点的树,让你选出k个黑点，其他都为白点。然后贡献为同颜色点之间两两路径和.最大化贡献。
$k\le n\le 2000$

题目思路:

1.首先定义状态

$dp[i][j]$为节点i，有j个黑点的最大值
这样以来，我们可以将一颗子树的黑点个数看作是背包的体积。然后二元组<子树中黑点个数,点对贡献>就代表着一个物品。对于每个子树来说，只能向根转移一种二元组(黑点个数0~min(子树大小,k)).

问题就转化为树上分组背包，体积恰好为k的最大价值.

2.计算点对贡献

由于这个题目中k是固定了的。所以对于一个子树x.若含有j个黑点，那么x的补集树就含有 k - j 个黑点，白点个数一样。所以贡献为:

$(j\ast (k-j)+(sz[x]-j)\ast (n-k-sz[x]+j))\ast w[i]$

注意：一定要控制循环上界为树的大小sz。这样才能保证每个点对只在它们的LCA处计算一次.复杂度为O(n^2).

AC代码(模板):

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
const int maxn = 2000 + 5;
const int mod = 1e9 + 7;
ll dp[maxn][maxn];
ll sz[maxn] , n;
vector<pii> e[maxn];
ll tmp[maxn] , up;
void dfs (int u , int fa)
{
   
    sz[u] = 1;
    for (auto g : e[u]){
   
        int v = g.first;
        ll w = g.second;
        if (v == fa) continue;
        dfs(v , u);
        for (int i = 0 ; i <= sz[u] + sz[v]; i++) tmp[i] = dp[u][i];
        for (int i = 0 ; i <= min(sz[u] , up) ; i++){
   
            for (int j = 0 ; j <= min(sz[v] , up) ; j++){
   
                if (i + j > up) continue;
                ll val = 1ll * j * (up - j) /*黑点*/ + 1ll * (sz[v] - j) * (n - sz[v] - (up - j));
                val *= w;
                tmp[i + j] = max (tmp[i + j] , dp[u][i] + dp[v][j] + val);
            }
        }
        for (int i = 0 ; i <= sz[u] + sz[v] ; i++) dp[u][i] = tmp[i];
        sz[u] += sz[v];
    }
    return ;
}
int main()
{
   
    ios::sync_with_stdio(false);
    cin >> n >> up;
    for (int i = 1 ; i < n ; i++){
   
        int x , y , z; cin >> x >> y >> z;
        e[x].pb (mp(y , z));
        e[y].pb (mp(x , z));
    }
    dfs (1 , 0);
    cout << dp[1][up] << endl;
    return 0;
}

二.洛谷P1272

题目大意:

给你一棵树，让你删除最少的边使得生成一个含有$P$个节点的连通块.

题目思路:

令 f (