【洛谷题解】P2015 二叉苹果树

题目链接:https://www.luogu.com.cn/problem/P2015
难度:普及+/提高
涉及知识点:树形DP

题意

给定一棵有 n n n 个节点的苹果树,对于一条连接 a a a b b b 的树枝上有 c c c 个苹果。要求剪去其中的一些树枝,最后留下 m m m 条树枝。求剪枝后最多能保留多少个苹果。

分析与解决

由“最多可保留多少苹果”,我们可以推断该题应使用动态规划求最大值。设状态为 f [ u ] [ j ] f[u][j] f[u][j],表示以 u u u 为根的树保留 j j j 条树枝后,最多可以留住的苹果数量。

将一个节点连向的各个子节点看做若干个物品组,这样问题就被转化为了分组背包问题。第一层循环物品组(也就是子节点),第二层循环体积,最大为 m m m,最小为 0,第三层循环保留的树枝个数(即决策)。

如果以 u u u 为根的子树内保留了 k k k 根树枝(不包含 u u u 与其子树节点 v v v 相连的边),那么就用 f [ u ] [ j − k − 1 ] + f [ v ] [ k ] + w [ u − > v ] f[u][j-k-1]+f[v][k]+w[u->v] f[u][jk1]+f[v][k]+w[u>v] 即可。为什么是 j − k − 1 j-k-1 jk1 呢?因为我们必须保留到根节点的那一条,换句话说,你想要在这棵子树内保留边,你就必须保留 u − > v u->v u>v 的边。

最后得到状态转移方程为:
f [ u ] [ j ] = m a x { f [ u ] [ j − k − 1 ] + f [ v ] [ k ] + w [ i ] } f[u][j]=max\left\{f[u][j-k-1]+f[v][k]+w[i]\right\} f[u][j]=max{f[u][jk1]+f[v][k]+w[i]}

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 110, M = 2 * N;

//f[i][j]表示以i为根的子树保留j根树枝,留住了多少苹果 

int n, m;
int h[N], e[M], ne[M], w[M], idx;
int f[N][N];

void add(int a, int b, int c)
{
	e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
}

void dfs(int u, int father)
{
	for (int i = h[u]; ~i; i = ne[i])
	{
		if (e[i] == father) continue;
		dfs(e[i], u);
		
		for (int j = m; j >= 0; j--)
		{
			for (int k = 0; k < j; k++)
			{
				f[u][j] = max(f[u][j], f[u][j - 1 - k] + f[e[i]][k] + w[i]);
			}
		} 
	}
} 

int main()
{
    cin >> n >> m;
	memset(h, -1, sizeof h);
	for (int i = 0; i < n - 1; i++)
	{
		int a, b, c;
		scanf("%d%d%d", &a, &b, &c);
		add(a, b, c);
		add(b, a, c);
	}
	
	
	dfs(1, -1);
	
	cout << f[1][m] << endl;
	
	return 0;
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

oier_Asad.Chen

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值