洛谷P2015 二叉苹果树树形dp

最新推荐文章于 2022-08-10 23:27:58 发布

SigmaQuadrant

最新推荐文章于 2022-08-10 23:27:58 发布

阅读量118

点赞数

分类专栏：比赛题解

本文链接：https://blog.csdn.net/qq_40859782/article/details/101055517

版权

比赛题解专栏收录该内容

218 篇文章 0 订阅

订阅专栏

这道题目要求在给定的树结构中，保留最多k条边，以最大化边权总和。问题转化为树形动态规划，其中fu,j表示节点u子树保留j条边的最大边权和。解题策略是采用容量为j的背包，对每个子树进行01背包操作，并逆序枚举j以优化转移过程。最终答案为f1,k，即根节点的子树保留k条边的最大边权和。" 51931081,5685103,WCDMA DPCH/EDCH功率控制计算解析,"['无线通信', '移动通信', '3G技术', '信号处理', 'WCDMA标准']

摘要由CSDN通过智能技术生成

给出 $n\leq100$ 个点的树，树上有边权。给出 $k\leq100$ ，求问保留 $k$ 条边权的最大值是多少。保留 $k$ 条边权指的是你要保留子树就必须要保留与此连接的边，也就是不能断开。
$f_{u,j}$ 表示 $u$ 这个子树保留 $j$ 条边的最大值。答案即为 $f_{1,k}$ 。转移的时候可以认为是对 $u$ 这个子树做一个容量 $j$ 的背包，对每个子树做01背包，因此对 $j$ 要逆序枚举。同时容量显然也不会超过子树大小 $- 1$ 。
转移方程 $f_{u,j}=max\{f_{u,j-k-1}+f_{v,k}+e_{w} \}$ 。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=105;
struct edge{ int v,w; };
vector<edge> go[N];
int f[N][N];
int sz[N];
int n,tot;
void dfs(int u,int fa) {
	f[u][0]=0;
	sz[u]=1;
	for(auto &e:go[u]) {
		int v=e.v,w=e.w;
		if(v==fa) continue; 
		dfs(v,u);
		sz[u]+=sz[v];
		for(int j=min(sz[u]-1,tot);j>=1;j--) {
			int up=min(j,sz[v]);
			for(int k=0;k<up;k++) {
				f[u][j]=max(f[u][j],f[u][j-k-1]+f[v][k]+w);
			}
		}
	}
}
int main() {
	scanf("%d%d",&n,&tot);
	for(int i=1;i<n;i++) {
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		go[u].push_back({v,w});
		go[v].push_back({u,w});
	}
	dfs(1,0);
	printf("%d\n",f[1][tot]); 
	return 0;
}