LightOJ 1094

最新推荐文章于 2019-01-15 18:58:31 发布

转载最新推荐文章于 2019-01-15 18:58:31 发布 · 87 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/9968jie/p/5803980.html

本文介绍了一种求解树直径的经典算法，即通过两次深度优先搜索（DFS）来找到树中任意两点间的最大距离。首先从任意一点出发找到离它最远的节点，再从此节点出发寻找离它最远的节点，这两点间的距离即为树的直径。

题意：就是求一个树的直径，也就是求任意两点的最大距离。

做法：跑两遍DFS，详见代码。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<cctype>
#include<queue>
#include<vector>
#include<algorithm>

using namespace std;
typedef long long LL;
#define N 100100
#define INF 0x3f3f3f3f

int n, k, head[N], dist[N], Max, Index;
struct node
{
    int v, len, next;
} maps[N<<2];

void add(int u, int v, int len)
{
    maps[k].v=v;
    maps[k].len=len;
    maps[k].next=head[u];
    head[u]=k++;
}

void DFS(int u, int len)
{
    dist[u]=len;

    if(dist[u]>Max)
    {
        Max=dist[u];
        Index=u;
    }

    for(int i=head[u]; i!=-1; i=maps[i].next)
    {
        int v=maps[i].v;
        if(dist[v]==-1)
            DFS(v, dist[u]+maps[i].len);
    }
}

int main()
{
    int T, a, b, c, cas=1;
    scanf("%d", &T);

    while(T--)
    {
        k=1;
        memset(head, -1, sizeof(head));
        scanf("%d", &n);
        for(int i=0; i<n-1; i++)
        {
            scanf("%d%d%d", &a, &b, &c);
            add(a, b, c);
            add(b, a, c);
        }

        Max=Index=-1;
        memset(dist, -1, sizeof(dist));
        DFS(0, 0);

        memset(dist, -1, sizeof(dist));
        DFS(Index, 0);
        printf("Case %d: %d\n", cas++, Max);
    }
    return 0;
}