树的中心

树的中心

题意

• 找到一个点使最远的点最小

思路

• 画出一颗树，找到一个点，这个点的最远距离，有俩种，一种是往下走，另外一种是往上走，往上走还有俩种，一种是再往上走另外一种是往下走，
• 因此，我们需要，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;
}