P1122 最大子树和

最新推荐文章于 2025-02-07 13:41:09 发布

yyd_s

最新推荐文章于 2025-02-07 13:41:09 发布

阅读量187

点赞数

分类专栏：树与图上的动态规划（洛谷）文章标签： c语言 c++

本文链接：https://blog.youkuaiyun.com/yyd_s/article/details/124072786

版权

树与图上的动态规划（洛谷）专栏收录该内容

4 篇文章

订阅专栏

该博客介绍了使用深度优先搜索（DFS）结合动态规划（DP）解决寻找图中最大子树价值的问题。代码示例展示了如何在C++中实现这一算法，通过遍历树形结构并递归计算每个节点的最大值。最终，程序输出整个树中具有最大和的子树的值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

//#define local
#include<cstdio>
#include<iostream>
#include<vector>
#include<cmath>
#define MAXN 16005
#define INF 0x7fffffff 
using namespace std;
typedef long long ll; 
int dp[MAXN];//dp[i]表示当前结点可得到的最大值 
vector<int>e[MAXN];
int n; 
void dfs(int cur,int fa){
  	for(int i=0;i<e[cur].size();++i){
		int next=e[cur][i];
		if(next!=fa){
		dfs(next,cur);
		dp[cur]=max(dp[cur],dp[cur]+dp[next]);
		}
	}
}
int main(){
#ifdef local
freopen("data.in","rb",stdin);
//freopen("data.out","wb",stdout);
#endif
scanf("%d",&n);
for(int i=1;i<=n;++i)
	scanf("%d",&dp[i]);
for(int i=1;i<=n-1;++i){
	int a,b;
	scanf("%d%d",&a,&b);
	e[a].push_back(b);
	e[b].push_back(a);
}
dfs(2,0);
int ans=-INF;
for(int i=1;i<=n;++i){
	if(ans<dp[i])
	ans=dp[i];
}
printf("%d",ans);
return 0;
}