树的直径新解法？

#include<bits/stdc++.h>
using namespace std;
const int maxn=5e5+5;
int n,cnt,ans=-1e9;
int head[maxn];
struct arr{
	int nxt,to,val;
}edge[maxn<<1];
inline void add(int x,int y,int z)
{
	edge[++cnt]=(arr){head[x],y,z};
	head[x]=cnt;
}
int dfs(int u,int fa)
{
	int mx_len=0;
	for (int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].to;
		if (v==fa)continue;
		int mx_v=dfs(v,u)+edge[i].val;
		ans=max(ans,mx_len+mx_v);
		mx_len=max(mx_len,mx_v);
	}
	return mx_len;
}
int main()
{
	scanf("%d",&n);
	for (int i=1,x,y,z;i<n;i++){
		scanf("%d%d%d",&x,&y,&z);add(x,y,z);add(y,x,z);
	}
	dfs(1,0);
	printf("%d\n",ans);
	return 0;
}