蓝桥杯生命之树(DP)

复制代码
 输入:5
     1 -2 -3 4 5
     4 2
     3 1
     1 2
     2 5
输出:8

思路:

无向无环图的存储,且节点编号从1 开始(遍历注意一下)。本题是求这颗树中相互连接的节点的和最大问题,也就是连通块和最大问题,用dfs遍历图,dp[x]表示以节点x为跟的部分的和,如果dp[v[x][i]]小于0,那么不会增大dp[x]的值,所以舍去,如果dp[[x][i]]>0,则dp[x]+=dp[[x][i]],最后在比较各部分的dp,找到最大值即为所求。

注意:0ll是表示长整型的0,max比较一般要同类型

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1e5+5;
int n,a[N],dp[N];
vector<int> v[N];
void dfs(int x,int fa) 
{
	dp[x]=a[x];
	for(int i=0;i<v[x].size();i++)
	{
		if(v[x][i]==fa) continue;
		dfs(v[x][i],x);
		dp[x]+=max(0ll,dp[v[x][i]]);//不能确定dp什么时候<0,因此要递归遍历
	}
}
signed main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	for(int i=0;i<n-1;i++)
	{
		int a,b;
		cin>>a>>b;
		v[a].push_back(b);
		v[b].push_back(a);
	}
	dfs(1,-1);
	int ans=-0x3f3f3f3f;
	for(int i=1;i<=n;i++)
	{
		ans=max(ans,dp[i]);
	}
	cout<<ans<<endl;
 } 
相关推荐
Trouvaille ~13 小时前
【优选算法篇】哈希表——空间换时间的极致艺术
c++·算法·leetcode·青少年编程·蓝桥杯·哈希算法·散列表
AlunYegeer14 小时前
面试问题controller和service能不能互相替换
面试·职场和发展
Epiphany.55614 小时前
题目 3146: 蓝桥杯2023年第十四届省赛真题-网络稳定性 时间限制: 1.5s 内存限制: 256MB
职场和发展·蓝桥杯
七禾页丫16 小时前
面试记录19 软件设计师
面试·职场和发展
無限進步D17 小时前
简单贪心算法 cpp
c++·算法·贪心算法·蓝桥杯·入门·竞赛
爱吃涮毛肚的肥肥(暂时吃不了版)17 小时前
Leetcode——181.超过经理收入的员工
算法·leetcode·职场和发展
醉颜凉17 小时前
Seal^_^【送书活动第8期】——《ChatGLM3大模型本地化部署、应用开发与微调》
人工智能·职场和发展·送书活动·chatglm3大模型
仰泳的熊猫17 小时前
题目2580:蓝桥杯2020年第十一届省赛真题-分类计数
数据结构·c++·算法·蓝桥杯
软件测试君17 小时前
自动化测试路线图之自动化测试完整指南
自动化测试·软件测试·测试工具·面试·职场和发展·单元测试·职场经验
_饭团18 小时前
C 语言数据存储全解析:原反补码、大小端与 IEEE 754 浮点数
c语言·数据结构·算法·leetcode·面试·蓝桥杯·学习方法