给定一棵树,每个结点和边均存在点权和边权,要求选中一个非空的连通块使得该连通块的权最大。
初见想法:枚举到当前结点时,若不选择与某一个儿子的连线,则该儿子将独立成为新的子树,维护多个树能够得到的最大值即可。细想发现:这不直接把状态 f [ i ] f[i] f[i] 设置为以 i i i 结点为根且选中 i i i 结点就可以转移了吗)
由此得到对于任一非叶子结点,如果某一儿子的贡献与两点间的边权之和不为负数,那么就可以选择将这一部分加入连通块中。转移方程为: f [ f a t h e r ] = f [ s o n ] + e d g e [ f a t h e r ] [ s o n ] f[father]=f[son]+edge[father][son] f[father]=f[son]+edge[father][son]
唯一坑点:一定要选一个非空的连通块,当所有权均为负数时得到的最优解仍然为负数,因此 a n s ans ans 的初值一定要足够小。
#include <bits/stdc++.h>
#define int long long
#define PII pair<int,int>
using namespace std;
const int N=1e5+10;
int a[N],f[N];
vector<PII> h[N];
/*
从根结点起进行遍历 选择与其儿子的连线则拥有该边和儿子
*/
void dfs(int u,int father){
f[u]+=a[u];
for(auto x:h[u]){
if(x.first==father)
continue;
dfs(x.first,u);
if(f[x.first]+x.second>=0)
f[u]+=f[x.first]+x.second;
}
}
signed main(){
int n,x,y,w;
cin>>n;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=1;i<n;i++){
cin>>x>>y>>w;
h[x].push_back({y,w});//first为连接的点 second为与该点间边的权值
h[y].push_back({x,w});
}
dfs(1,0);
int ans=-1e18;
for(int i=1;i<=n;i++){
ans=max(ans,f[i]);
}
cout<<ans<<endl;
return 0;
}