树的直径

板子

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 1e6 + 5;
const int inf = 0x3f3f3f3f;
const ll MOD = 1e9 + 7;
ll dis[maxn];
int vis[maxn];
int head[maxn],cnt;
struct node{ int v; ll w; int next;
}e[maxn*2];//双向边
void add(int u,int v,ll w){
	e[cnt].v=v;
	e[cnt].w=w;
	e[cnt].next=head[u];
	head[u]=cnt++;
}
void dfs(int u){
	vis[u]=1;
	for(int i=head[u];~i;i=e[i].next){
		int v=e[i].v;
		if(!vis[v]){
			dis[v]=dis[u]+e[i].w;
			dfs(v);
		}
	}
} 
int main(){
	memset(vis,0,sizeof(vis));
	memset(dis,inf,sizeof(dis));
	memset(head,-1,sizeof(head));
	int n; cin>>n;
	for(int i=1;i<n;i++){
		int x,y; ll z; cin>>x>>y>>z;
		add(x,y,z); add(y,x,z);
	}
	dis[1]=0;
	dfs(1);//从任意一个点出发找第一个端点
	ll mx=-inf; int id=0;
	for(int i=1;i<=n;i++){
		if(dis[i]>=mx&&dis[i]!=inf){
			mx=dis[i];
			id=i;
		}
	} 
	memset(vis,0,sizeof(vis));
	memset(dis,inf,sizeof(dis));
	int P=id;//第一个端点
	mx=-inf; id=0; 
	dis[P]=0;//自己到自己为0
	dfs(P);
	for(int i=1;i<=n;i++){
		if(dis[i]>=mx&&dis[i]!=inf){
			mx=dis[i];
			id=i;
		}
	}
	int Q=id;//第二个端点
	cout<<P<<" "<<Q<<" "<<mx<<endl; 
	//mx即两端点间的距离，即是树的直径 
	 
}