Acwing 1072 求树的直径(dfs || 树形dp)

题目
题意: 求树的直径，但是可能有负边权，用两遍dfs不行。
思路: 树形dp. 以i为树根的最大路径为经过i的最大路 + 次大路，只要我们求出max{以每个点为树根的最大路+次大路}即可得到答案.
时间复杂度: O(n)
代码:

#include<bits/stdc++.h>
using namespace std;
#define mem(a,x) memset(a,x,sizeof(a))
const int N = 1e4+10;
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 ans = 0;
int n,m,k,T;
int dfs(int cur,int fa)
{
	// int dist = 0; //以cur为树根的最大距离
	int d1 = 0,d2 = 0;
	for(int i=h[cur];~i;i=ne[i])
	{
		int j = e[i];
		if(j == fa) continue;
		int d = dfs(j,cur) + w[i];
		if(d >= d1) d2 = d1,d1 = d;
		else if(d > d2) d2 = d;
	}
	ans = max(ans,d1 + d2);
	return d1;
}
void solve()
{
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	cin>>n; mem(h,-1);
	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);
	cout<<ans;
}
signed main(void)
{
	solve();
	return 0;
}