牛客手速月赛45 E(还就内个遇到dp就不会，都怪开会，难蚌)

题目
题意: 打着打着比赛被喊出去开会，组织生活会确实很重要，但是我还没见过哪个学院天天开这么多会的，计算机开会学院是吧?班长团支书一个专业前5都没有，什么成分，不多说了。麻了，还是自己太菜，不能怪外部因素，人家法学院学姐一手rank1/234+团支书+中共党员+六级631，我与人家的唯一共同点是均为团支书，难蚌。窥镜而自视，又弗如远甚。不能动气，动气又牙疼了。封校没法去医院拔智齿，难蚌。
  回到题目，给定n个点的树，每个点有对应的点权。边权可能为负。找到一个最大贡献的非空连通块。
  贡献: 连通块种的点权和+边权和
思路: 树形dp，大概是。基本上是很基础的树形dp了，直接dfs乱写。
f[i]:以i为根的最大贡献
f[i] += f[j] + w[i]
时间复杂度: O(n)
代码:

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
#define mem(a,x) memset(a,x,sizeof(a))
int n,m,k,T;
int h[N],e[N<<1],ne[N<<1],w[N<<1],idx = 0;
void add(int a,int b,int c)
{
	e[idx] = b,w[idx] = c,ne[idx] = h[a],h[a] = idx++; 
}
int a[N];
int f[N]; //以i为根的最大值
void dfs(int u,int fa)
{
	f[u] = a[u];
	for(int i=h[u];~i;i=ne[i])
	{
		int j = e[i];
		if(j == fa) continue;
		dfs(j,u);
		int t = f[j] + w[i];
		if(t >= 0)
		{
			f[u] += t;
		}
	}
}
void solve()
{
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	cin>>n;
	mem(h,-1); idx = 0;
	for(int i=1;i<=n;++i) cin>>a[i];
	for(int i=0;i<n-1;++i)
	{
		int x,y,z; cin>>x>>y>>z;
		add(x,y,z),add(y,x,z);
	}
	dfs(1,0);
	// int ans = a[1];
	// for(int i=1;i<=n;++i) ans = max(ans,f[i]);
	cout<<f[1];
}
signed main(void)
{
	solve();
	return 0;
}