蓝桥杯双周赛2

串门

#include 
using namespace std;

const int N = 2e5+10;
typedef long long ll;
ll h[N],dis[N];
ll cnt,n;
struct edge
{
    ll to,w,next;
}e[N];

void add(ll from,ll to,ll w)
{
    e[++cnt].w=w;
    e[cnt].to=to;
    e[cnt].next=h[from];
    h[from]=cnt;
}

void dfs(ll u,ll father,ll distance)
{
    dis[u]=distance;
    for(ll i=h[u];i;i=e[i].next)
    {
        ll j=e[i].to;
        if(j!=father)dfs(j,u,distance+e[i].w);
    }
}

int main()
{
    cin>>n;
    ll sum=0;
    for(int i=0;i>a>>b>>c;
        add(a,b,c);
        add(b,a,c);
        sum+=2*c;
    }
    dfs(1,0,0);

    int u=1;
    for(int i=2;i<=n;i++)
        if(dis[i]>dis[u])u=i;
    dfs(u,0,0);
    u=1;
    for(int i=2;i<=n;i++)
        if(dis[i]>dis[u])u=i;
    cout<

你可能感兴趣的:(算法,深度优先,图论)