Java教程

树的中心

本文主要是介绍树的中心,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int maxn=1e5+10;
vector<pair<int,int>>g[maxn];
int d1[maxn],d2[maxn];
int p1[maxn],up[maxn];
int dfs_down(int u,int fa){
    for(auto &t:g[u]){
        int x=t.first,y=t.second;
        if(x==fa)continue;
        int z=dfs_down(x,u)+y;
        if(z>=d1[u])d2[u]=d1[u],d1[u]=z,p1[u]=x;
        else if(z>=d2[u])d2[u]=z;
    }
    return d1[u];
}
void dfs_up(int u,int fa){
    for(auto &t:g[u]){
        int x=t.first,y=t.second;
        if(x==fa)continue;
        if(x==p1[u])up[x]=max(up[u],d2[u])+y;
        else up[x]=max(up[u],d1[u])+y;
        dfs_up(x,u);
    }
}
int main()
{
    ios::sync_with_stdio(false);
    int n;cin>>n;
    for(int i=1,u,v,val;i<n;i++){
        cin>>u>>v>>val;
        g[u].push_back(make_pair(v,val));
        g[v].push_back(make_pair(u,val));
    }
    dfs_down(1,-1);
    dfs_up(1,-1);
    int ans=0x3f3f3f3f;
    for(int i=1;i<=n;i++)
    ans=min(ans,max(d1[i],up[i]));
    cout<<ans;
    return 0;
}

  

这篇关于树的中心的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!