E-筑巢（树形dp）

传送门

给定一棵树，每个结点和边均存在点权和边权，要求选中一个非空的连通块使得该连通块的权最大。

初见想法：枚举到当前结点时，若不选择与某一个儿子的连线，则该儿子将独立成为新的子树，维护多个树能够得到的最大值即可。细想发现：这不直接把状态 $f[i]$ 设置为以 $i$ 结点为根且选中 $i$ 结点就可以转移了吗）

由此得到对于任一非叶子结点，如果某一儿子的贡献与两点间的边权之和不为负数，那么就可以选择将这一部分加入连通块中。转移方程为：$f[father]=f[son]+edge[father][son]$

唯一坑点：一定要选一个非空的连通块，当所有权均为负数时得到的最优解仍然为负数，因此 $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;
}