码迷,mamicode.com
首页 > 其他好文 > 详细

LightOJ 1094

时间:2016-08-24 19:01:53      阅读:170      评论:0      收藏:0      [点我收藏+]

标签:

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

做法:跑两遍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;
}

 

LightOJ 1094

标签:

原文地址:http://www.cnblogs.com/9968jie/p/5803980.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!