题目:[HAOI2015]树上染色
分析:
dp[i][j]表示第i个点染j个黑色时的对答案的最大贡献。
这里我们采用“合并法”求dp,即将两个子问题合并算大问题的答案。
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;//三年竞赛一场空,不开long long见祖宗
//typedef __int128 lll;
#define print(i) cout << "debug: " << i << endl
#define close() ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
#define mem(a, b) memset(a, b, sizeof(a))
#define pb(a) push_back(a)
#define x first
#define y second
typedef pair<int, int> par;
const ll mod = 10007;
const int maxn = 2000 + 10;
const int inf = 0x3f3f3f3f;
struct edge
{
int v, val, nex;
edge(int v = 0, int val = 0, int nex = 0) : v(v), val(val), nex(nex){}
}e[maxn << 1];
int head[maxn], tot;
ll siz[maxn];
ll dp[maxn][maxn];
int n, k;
void init() {mem(head, -1), tot = 0;}
void addedge(int x, int y, int val)
{
e[tot] = edge(y, val, head[x]);
head[x] = tot++;
}
void dfs(int u, int fa)
{
siz[u] = 1;
for(int i = head[u]; ~i; i = e[i].nex)
{
int v = e[i].v;
if(v == fa) continue;
dfs(v, u);
for(int j = siz[u]; j >= 0; j--)
for(int p = siz[v]; p >= 0; p--)//合并之前保证两个子问题是没有交集的。
{
ll val = (1ll * p * (k - p) + 1ll * (siz[v] - p) * (n - k - siz[v] + p)) * e[i].val;
dp[u][j + p] = max(dp[u][j + p], dp[u][j] + dp[v][p] + val);
}
siz[u] += siz[v];//合并完了再更新siz
}
}
int main()
{
cin >> n >> k;
init();
for(int i = 1; i < n; i++)
{
int a, b, c; cin >> a >> b >> c;
addedge(a, b, c), addedge(b, a, c);
}
dfs(1, -1);
cout << dp[1][k] << endl;
}