关闭

hihocoder-1050 树中的最长路(dfs)

标签: hihocoderDFS
1092人阅读 评论(0) 收藏 举报
分类:

时间限制:10000ms
单点时限:1000ms
内存限制:256MB

描述

上回说到,小Ho得到了一棵二叉树玩具,这个玩具是由小球和木棍连接起来的,而在拆拼它的过程中,小Ho发现他不仅仅可以拼凑成一棵二叉树!还可以拼凑成一棵多叉树——好吧,其实就是更为平常的树而已。

但是不管怎么说,小Ho喜爱的玩具又升级换代了,于是他更加爱不释手(其实说起来小球和木棍有什么好玩的是吧= =)。小Ho手中的这棵玩具树现在由N个小球和N-1根木棍拼凑而成,这N个小球都被小Ho标上了不同的数字,并且这些数字都是出于1..N的范围之内,每根木棍都连接着两个不同的小球,并且保证任意两个小球间都不存在两条不同的路径可以互相到达。总而言之,是一个相当好玩的玩具啦!

但是小Hi瞧见小Ho这个样子,觉得他这样沉迷其中并不是一件好事,于是寻思着再找点问题让他来思考思考——不过以小Hi的水准,自然是手到擒来啦!

于是这天食过早饭后,小Hi便对着又拿着树玩具玩的不亦乐乎的小Ho道:“你说你天天玩这个东西,我就问你一个问题,看看你可否知道?”

“不好!”小Ho想都不想的拒绝了。

“那你就继续玩吧,一会回国的时候我不叫上你了~”小Hi严肃道。

“诶!别别别,你说你说,我听着呢。”一向习惯于开启跟随模式的小Ho忍不住了,马上喊道。

小Hi满意的点了点头,随即说道:“这才对嘛,我的问题很简单,就是——你这棵树中哪两个结点之间的距离最长?当然,这里的距离是指从一个结点走到另一个结点经过的木棍数。”。

“啊?”小Ho低头看了看手里的玩具树,困惑了。

输入

每个测试点(输入文件)有且仅有一组测试数据。

每组测试数据的第一行为一个整数N,意义如前文所述。

每组测试数据的第2~N行,每行分别描述一根木棍,其中第i+1行为两个整数Ai,Bi,表示第i根木棍连接的两个小球的编号。

对于20%的数据,满足N<=10。

对于50%的数据,满足N<=10^3。

对于100%的数据,满足N<=10^5,1<=Ai<=N, 1<=Bi<=N

小Hi的Tip:那些用数组存储树边的记得要开两倍大小哦!

输出

对于每组测试数据,输出一个整数Ans,表示给出的这棵树中距离最远的两个结点之间相隔的距离。

样例输入
8
1 2
1 3
1 4
4 5
3 6
6 7
7 8
样例输出
6

#include <cstdio>
#include <cstring>

using namespace std;

struct Node {
    int e,nxt;
}node[200005];//数组模拟链表

int n,num,maxDep,nd;
int h[100005];

void add(int s,int e) {//加边
    node[++num].e=e;
    node[num].nxt=h[s];
    h[s]=num;
}

void dfs(int dep,int u,int pre) {
    if(dep>maxDep) {//如果该路径的长度大于maxDep,更新maxDep和最长路的端点
        maxDep=dep;
        nd=u;
    }
    int i=h[u];
    while(i) {
        if(node[i].e!=pre)//因为是树,所以只用判断当前节点的前一个节点是否与其下一个节点相同
            dfs(dep+1,node[i].e,u);
        i=node[i].nxt;
    }
}

int main() {
    int s,e,i;
    while(1==scanf("%d",&n)) {
        memset(h,0,sizeof(h));
        num=0;
        for(i=1;i<n;++i) {
            scanf("%d%d",&s,&e);
            add(s,e);
            add(e,s);
        }
        maxDep=0,nd=0;
        dfs(0,1,0);//找到最长路的一个端点
        maxDep=0;
        dfs(0,nd,0);//找到最长路的长度
        printf("%d\n",maxDep);
    }
    return 0;
}


法二:

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

struct Node {
    int e,nxt;
}node[200005];//数组模拟链表

int n,num,ans;
int h[100005];

void add(int s,int e) {//加边
    node[++num].e=e;
    node[num].nxt=h[s];
    h[s]=num;
}

int dfs(int u,int pre) {
    int i=h[u],d,fir=0,sec=0;
    while(i) {
        if(node[i].e!=pre) {//因为是树,所以只用判断当前节点的前一个节点是否与其下一个节点相同
            d=dfs(node[i].e,u)+1;
            if(d>fir)//如果子树的本次最长路径大于目前最长路径,更新最长路径长度和次长路径长度
                sec=fir,fir=d;
            else if(d>sec)//如果子树的本次最长路径仅大于目前次长路径,更新次长路径长度
                sec=d;
        }
        i=node[i].nxt;
    }
    ans=max(ans,fir+sec);//更新树的直径
    return fir;
}

int main() {
    int s,e,i;
    while(1==scanf("%d",&n)) {
        memset(h,0,sizeof(h));
        ans=num=0;
        for(i=1;i<n;++i) {
            scanf("%d%d",&s,&e);
            add(s,e);
            add(e,s);
        }
        dfs(1,0);
        printf("%d\n",ans);
    }
    return 0;
}





1
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:103563次
    • 积分:3804
    • 等级:
    • 排名:第8412名
    • 原创:288篇
    • 转载:0篇
    • 译文:0篇
    • 评论:37条
    最新评论