一.洛谷P3177
题目大意:
给你一颗含n个点的树,让你选出k个黑点,其他都为白点。然后贡献为同颜色点之间两两路径和.最大化贡献。
k ≤ n ≤ 2000 k \leq n \leq 2000 k≤n≤2000
题目思路:
1.首先定义状态
d p [ i ] [ j ] dp[i][j] dp[i][j]为节点i,有j个黑点的最大值
这样以来,我们可以将一颗子树的黑点个数看作是背包的体积。然后二元组<子树中黑点个数,点对贡献>就代表着一个物品。对于每个子树来说,只能向根转移一种二元组(黑点个数0~min(子树大小,k)).
问题就转化为树上分组背包,体积恰好为k的最大价值.
2.计算点对贡献
由于这个题目中k是固定了的。所以对于一个子树x.若含有j个黑点,那么x的补集树就含有 k - j 个黑点,白点个数一样。所以贡献为:
( j ∗ ( k − j ) + ( s z [ x ] − j ) ∗ ( n − k − s z [ x ] + j ) ) ∗ w [ i ] (j * (k - j) + (sz[x] - j)*(n - k -sz[x]+j))*w[i] (j∗(k−j)+(sz[x]−j)∗(n−k−sz[x]+j))∗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 P P个节点的连通块.
题目思路:
令 f (