©2020-2024 ioDraw All rights reserved,
Privacy Policy
数据范围
1≤n≤10^5,
每个节点的评分的绝对值均不超过 10^6。
输入样例：
5 1 -2 -3 4 5 4 2 3 1 1 2 2 5
输出样例：
8
分析：这道题是要我们在树中求一个最大连通块，我们可以定义f[i]为以i为根的子树中最大连通块的值
，这样结果就是f[1~n]中的最大值，树形DP过程比较简单，一开始令f[i]=w[i]，也就是令这个连通块只包含自己这一个点，然后只要以子节点为根的子树中最大连通块的值大于0，就加上，按照这样进行dp就可以求出答案，细节参照代码：
#include<cstdio> #include<iostream> #include<cstring> #include<vector>
#include<algorithm> #include<map> #include<cmath> #include<queue> using
namespace std; #define int long long const int N=1e5+10; int
w[N*2],f[N],e[N*2],ne[N*2],h[N],idx;//f[i]表示以i为根的连通块中的最大值 void add(int x,int y)
{ e[idx]=y; ne[idx]=h[x]; h[x]=idx++; } int dp(int x,int father) { f[x]=w[x];
for(int i=h[x];i!=-1;i=ne[i]) { int j=e[i]; if(j==father) continue;
f[x]+=max(dp(j,x),(int)0); } return f[x]; } signed main() { int n; cin>>n;
for(int i=1;i<=n;i++) scanf("%lld",&w[i]); memset(h,-1,sizeof h); for(int
i=1;i<n;i++) { int u,v; scanf("%lld%lld",&u,&v); add(u,v);add(v,u); } dp(1,0);
int ans=-0x3f3f3f3f; for(int i=1;i<=n;i++) ans=max(ans,f[i]);
printf("%lld",ans); return 0; }