Tyvj P1520 樹的直徑

http://www.tyvj.cn/p/1520

題目大意：求樹的直徑

題目分析：詳見本欄目總結。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
#include<cmath>
#include<cctype>
#include<cassert>
#include<climits>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define RepD(i,n) for(int i=n;i>=0;i--)
#define MEM(a) memset(a,0,sizeof(a))
#define MEMI(a) memset(a,127,sizeof(a))
#define MEMi(a) memset(a,128,sizeof(a))
#define INF (2139062143)
#define phiF (1000000006)
#define MAXN (1000000+10)
typedef long long LL;

struct info{
	int to,val,next;
}e[20005];

int first[10005],n,m,tot,f[10005],g[10005],ans,x,y,z;

void add(int x,int y,int z){
	 e[++tot].to=y;
	 e[tot].val=z;
	 e[tot].next=first[x];
	 first[x]=tot;
}

int dfs(int u,int fa){
	for (int p=first[u];p;p=e[p].next){
		int v=e[p].to;
		if (v!=fa){
			int tmp=e[p].val+dfs(v,u);
			if (tmp>f[u]){
				g[u]=f[u];
				f[u]=tmp;
			}else if (tmp>g[u]){
				g[u]=tmp; 
			} 
		}
		
	}
	ans=max(ans,f[u]+g[u]);
	return f[u];
	
}

int main(){
	scanf("%d",&n);
	For (i,n-1){
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
		add(y,x,z);
	}
	dfs(1,0);
    printf("%d",ans);
    
}