2020杭电多校第五场 1007[HDU6820] Tree

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6820

树形DP题。
在此之前并没有接触过树形DP，不得不借鉴别人的代码。在此基础上进行了详细的备注说明。
代码来源：https://www.cnblogs.com/codetogether/p/13441866.html

题解：
dp[i][0]表示当前点最多选k-1个儿子，考虑到父亲结点，这样每个点的度数不大于K，在此情况下所有儿子贡献最大值(不包括父亲)。
每次相当于获得一个子树的贡献。这种求法必定使得第一个dfs的点的度数不为K，其余点都满足了要求。
接下来用第二种dfs2求答案时，每次深入dfs2去找儿子结点度数大于K的情况时只要把当前点变为度数为K(除非儿子不够)即可。
dp[i][1]表示当前点往最多的儿子去选，则其他点必须仍受限制使得度数不大于K，在此情况下的所有儿子贡献最大值(将父亲转化为儿子)。
因为答案必定从dp[i][1]中取得，故要求实现把每个dp[i][1]中的i都看作是根结点。

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 2e5 + 5;
struct edge { int v; ll w; };
struct node { ll dp; int id; };
int q, n, k, fa[maxn];
ll dp[maxn][2], ans;
vector<edge>G[maxn];
void init() {
	for (int i = 1; i <= n; i++) {
		G[i].clear();
		dp[i][0] = dp[i][1] = 0;
	}
	ans = 0;
}
bool cmp(node x, node y) { return x.dp > y.dp; }
//dp[i][0]表示第i个结点最多选k-1个儿子的答案
//dp[i][1]表示第i个结点可以超过k条边,而其他点不行的答案
void dfs(int u) {//该点最多选k-1个儿子的情况
	ll* son = new ll[G[u].size() + 5];
	int cnt = 0;
	for (int i = 0; i < G[u].size(); i++) {
		int v = G[u][i].v; ll w = G[u][i].w;
		if (fa[u] == v)continue;
		fa[v] = u;
		dfs(v);
		son[cnt++] = dp[v][0] + w;//第cnt个儿子的贡献
	}
	sort(son, son + cnt);
	int x = cnt - k + 1; x = x > 0 ? x : 0;//加上贡献大的儿子,最多k-1个
	for (int i = cnt - 1; i >= x; i--)dp[u][0] += son[i];
	delete[]son;
}
void dfs2(int u) {//认为该点选的儿子数大于k-1的情况
	node* son = new node[G[u].size() + 5]; int cnt = 0;
	for (int i = 0; i < G[u].size(); i++) {
		int v = G[u][i].v; ll w = G[u][i].w;
		if (fa[u] != v)son[cnt++].dp = dp[v][0] + w;//v不是u的父亲,则贡献直接为dp[v][0](以v为根)+新的边权
		else son[cnt++].dp = dp[v][0] - dp[u][0];//v是u的父亲,由于dp[v][0]包含了dp[u][0]的贡献
		son[cnt - 1].id = v;					//反过来将u看作v的父亲,则v作为儿子的贡献为dp[v][0]-dp[u][0]
	}
	sort(son, son + cnt, cmp);
	for (int i = 0; i < cnt; i++)dp[u][1] += son[i].dp;//所有儿子都选上
	ans = max(ans, dp[u][1]);
	dp[u][0] = 0;//以下操作保证在对下一个点求度数大于k时,其父亲的度数为k(除非儿子不够)
	for (int i = 0; i < min(cnt, k - 1); i++)dp[u][0] += son[i].dp;//先选min(cnt,k-1)个贡献最大的儿子,目前度数为min(cnt,k-1)
	for (int i = 0; i < cnt; i++)if (son[i].id != fa[u]) {//然后再看情况,如果遍历儿子结点时,使得当前点凑成度数为k
		if (i >= k - 1)dp[u][0] += son[i].dp;//没加上这个儿子的贡献,dfs2后要使得该点的度数为k,因此要加上这个儿子
		else { if (k <= cnt)dp[u][0] += son[k - 1].dp; }//在dfs2后该点的度数为k-1,若能够再添度数应该加上第son[k-1],使得度数为k
		dfs2(son[i].id);
		if (i >= k - 1)dp[u][0] -= son[i].dp;
		else { if (k <= cnt)dp[u][0] -= son[k - 1].dp; }
	}
	delete[]son;
}
int main(void) {
	scanf("%d", &q);
	while (q--) {
		init();
		scanf("%d %d", &n, &k);
		for (int i = 0; i < n - 1; i++) {
			int u, v; ll w;
			scanf("%d %d %lld", &u, &v, &w);
			G[u].push_back(edge{ v,w });
			G[v].push_back(edge{ u,w });
		}
		if (k == 0) { printf("0\n"); continue; }
		dfs(1); dfs2(1);
		//先求出每个点最多只选k-1个儿子的情况,然后分别对每个点求当前点度数大于k的情况
		printf("%lld\n", ans);
	}
	return 0;
}