树的中心
题意
思路
- 画出一颗树,找到一个点,这个点的最远距离,有俩种,一种是往下走,另外一种是往上走,往上走还有俩种,一种是再往上走另外一种是往下走,
- 因此,我们需要,dfs出每个点到叶子节点的最大值,
- 再dfs出每个点往上走的最大值
代码
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
const int N=2e5;
int e[N],ne[N],w[N],h[N],idx;
int n;
int d1[N],d2[N],up[N],p1[N];
int is_left[N];
void add(int a,int b,int c)
{
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int dfs_1(int u,int father)
{
int dist=0;
d1[u]=-0x3f3f3f3f,d2[u]=-0x3f3f3f3f;
for(int i=h[u];~i;i=ne[i])
{
int j=e[i];
if(j==father) continue;
int d=dfs_1(j,u)+w[i];
dist=max(dist,d);
if(d>d1[u]){
d2[u]=d1[u];
d1[u]=d;
p1[u]=j;
}
else if(d>d2[u]) {
d2[u]=d;
}
}
if(d1[u]==-0x3f3f3f3f)
{
d1[u]=d2[u]=0;
is_left[u]=1;
}
return dist;
}
void dfs_2(int u,int father)
{
for(int i=h[u];~i;i=ne[i])
{
int j=e[i];
if(j==father ) continue;
if(p1[u]==j)
up[j]=max(up[u],d2[u])+w[i];
else up[j]=max(up[u],d1[u])+w[i];
dfs_2(j,u);
}
}
int main()
{
cin>>n;
memset(h,-1,sizeof h);
for(int i=1;i<n;i++)
{
int a,b,c;
cin>>a>>b>>c;
add(a,b,c),add(b,a,c);
}
dfs_1(1,-1);//
dfs_2(1,-1);
int res=d1[1];
for(int i=2;i<=n;i++)
{
if(is_left[i]) res=min(res,up[i]);
else res=min(res,max(d1[i],up[i]));
}
cout<<res<<endl;
return 0;
}