【题解】LuoGu2015：二叉苹果树

树型DP与背包算法

最新推荐文章于 2020-08-13 15:51:55 发布

原创最新推荐文章于 2020-08-13 15:51:55 发布 · 333 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#题解 #LuoGu #DP #树相关

题解同时被 3 个专栏收录

435 篇文章

订阅专栏

LuoGu

255 篇文章

订阅专栏

125 篇文章

订阅专栏

本文分享了一道树型DP的例题解析，通过O(n)遍历进行O(n^2)的背包算法优化，实现了高效求解。文章详细介绍了从输入读取到树状结构建立，再到动态规划的过程，最终输出最优解。

原题传送门
这应该是树型dp的例题，我竟然才做
现在的水平直接秒杀掉这道题目啦

$O (n)$ 遍历，过程中做 $O(n^2)$ 的背包，就好了， $ans=dp_{1,Q}$
注意一些小细节

我记得神仙 $k b l a c k$ 讲过 $O(n^2)$ 的做法，但是我忘了，只能 $O(n^3)$ 了
Code：

#include <bits/stdc++.h>
#define maxn 110
using namespace std;
struct Edge{
	int to, next, len;
}edge[maxn << 1];
int num, head[maxn], n, m, size[maxn], dp[maxn][maxn], ans;

inline int read(){
	int s = 0, w = 1;
	char c = getchar();
	for (; !isdigit(c); c = getchar()) if (c == '-') w = -1;
	for (; isdigit(c); c = getchar()) s = (s << 1) + (s << 3) + (c ^ 48);
	return s * w;
}

void addedge(int x, int y, int z){ edge[++num] = (Edge){y, head[x], z}, head[x] = num; }

void dfs(int u, int pre){
	for (int i = head[u]; i; i = edge[i].next){
		int v = edge[i].to;
		if (v != pre){
			dfs(v, u);
			size[u] += size[v] + 1;
			for (int j = min(m, size[u]); j; --j)
				for (int k = 0; k <= min(j - 1, size[v]); ++k)
					dp[u][j] = max(dp[u][j], dp[v][k] + dp[u][j - k - 1] + edge[i].len);
		}
	}
}

int main(){
	n = read(), m = read();
	for (int i = 1; i < n; ++i){
		int x = read(), y = read(), z = read();
		addedge(x, y, z), addedge(y, x, z);
	}
	dfs(1, 0);
	printf("%d\n", dp[1][m]);
	return 0;
}