树求直径

#include<cstdio>
#include<cstring>
#include<vector>

using namespace std;

int n,vis[10000],size[10000],dis[10000];
struct egde{
	int v,w;
};
vector<egde> g[10000];
int maxx=0,max1=0;

void addegde(int u,int v,int w)
{
	egde e1;e1.v=v;e1.w=w;
	g[u].push_back(e1);
	egde e2;e2.v=u;e2.w=w;
	g[v].push_back(e2);
}

int read()
{
	int f=1,x=0;
	char ch;
	do
	{
		ch=getchar();
		if(ch=='-')
		f=-1;
	}while(ch<'0' || ch>'9');
	do
	{
		x=x*10+ch-'0';
		ch=getchar();
	}while(ch>='0'&&ch<='9');
	return x*f;
}

void dfs(int u)
{
	vis[u]=1;
	for(int i=0;i<g[u].size();i++)
	{
		int v=g[u][i].v;
		if(vis[v]==1)continue;
		vis[v]=1;
		dis[v]=dis[u]+g[u][i].w;
		if(dis[v]>max1)
		{
			max1=dis[v]; 
			maxx=v;
		}
		dfs(v);
	}
}

int main(void){
	n=read();
	for(int i=1;i<n;i++)
	{
		int u=read(),v=read(),w=read();
		addegde(u,v,w);
	}
	dfs(1);
	int t=maxx;
	max1=0;
	memset(vis,0,sizeof(vis));
	memset(dis,0,sizeof(dis));
	dfs(t);
	int w=maxx;
	printf("%d %d %d",t,w,max1);
}