hihoCoder1050— 树中的最长路(深搜)

来源:互联网 发布:淘宝寄到海外 编辑:程序博客网 时间:2024/06/06 11:45

题目链接:传送门

时间限制: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,表示给出的这棵树中距离最远的两个结点之间相隔的距离。

样例输入
81 21 31 44 53 66 77 8
样例输出
6

解题思路:找到某结点u(始点)的子树中的最长路的终点v,则v是树中某最长路的一个端点(大概想一下就清楚了),再通过结点v向下搜索出最长的路径就是树中的最长路,写两个搜索就ok啦。


#include <cstdio>  #include <cstring>  #include <cmath>  #include <iostream>  #include <queue>#include <set>#include <string>#include <stack>#include <algorithm>#include <map>#include <bitset>using namespace std;  typedef long long ll;const int N = 100008;const int M = 100000000;const int INF = 0x3fffffff;const int mod = 1e9+7;const double Pi = acos(-1.0);const double sm = 1e-9;struct Edge{int nd;Edge*next;}m_edge[N*2];Edge*head[N];int vis[N],dist[N],Ecnt;void init(){Ecnt = 0;fill( head , head+N , (Edge*)0 );}void mkEdge( int a , int b ){m_edge[Ecnt].nd = b;m_edge[Ecnt].next = head[a];head[a] = m_edge+Ecnt++;}void dfs1( int u ){vis[u] = 1;int ans = 0;for( Edge*p = head[u] ; p ; p = p->next ){int v = p->nd;if( !vis[v] ){dist[v] = dist[u]+1;dfs1(v);}}}int dfs2( int u ){vis[u] = 1;for( Edge*p = head[u] ; p ; p = p->next ){int v = p->nd;if( !vis[v] ){dist[u] = max( dist[u] , dfs2(v)+1 );}}return dist[u];}int main(){int n,a,b;while( cin >> n ){//if( n == 1 ){ cout << 0 << endl; continue; }init();for( int i = 0 ; i < n-1 ; ++i ){cin >> a >> b;mkEdge(a,b);mkEdge(b,a);}fill( vis , vis+N , 0 );fill( dist , dist+N , 0 );dfs1( 1 );int node,len=0;for( int i = 1 ; i <= n ; ++i ){if( len < dist[i] ){len = dist[i];node = i;}}fill( vis , vis+N , 0 );fill( dist , dist+N , 0 );int ans = dfs2( node );cout << ans << endl;}return 0;}