链接:​​https://ac.nowcoder.com/acm/problem/17362​​​ 来源:牛客网
 

题目描述

    桃花一簇开无主,可爱深红映浅红。

                                        ——《题百叶桃花》

    桃花长在桃树上,树的每个节点有一个桃花,调皮的HtBest想摘尽可能多的桃花。HtBest有一个魔法棒,摘到树上任意一条链上的所有桃花,由于HtBest法力有限,只能使用一次魔法棒,请求出Htbest最多可以摘到多少个桃花。

输入描述:

第一行有一个正整数n,表示桃树的节点个数。
接下来n-1行,第i行两个正整数ai,bi ,表示桃树上的节点ai,bi之间有一条边。

输出描述:

第一行一个整数,表示HtBest使用一次魔法棒最多可以摘到多少桃花。

示例1

输入

复制

3
1 2
2 3

输出

复制

3

示例2

输入

复制

3
1 2
1 3

输出

复制

3

示例3

输入

复制

4
1 2
2 3
3 4

输出

复制

4

备注:

对于100%的测试数据:
1 ≤ n ≤ 1000000
数据量较大,注意使用更快的输入输出方式。

 题解 :两遍dfs , 第一遍从任意一个顶点开始 ,找到距离这个点最远的点,它一定是直径的一端,然后再从这个点开始dfs,找到最远的距离 . 

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define maxn 1000005
using namespace std;
struct Edge{
int next ; // 临界点
int u ;
int to ; // 终点

}edge[2*maxn];
int cnt ;
int maxs ;
int maxi ;
int vis[2*maxn] ;
int head[2*maxn] ;// 一般初始化为 -1
void Add(int u , int v )
{
edge[cnt].u = u ;
edge[cnt].next = head[u] ;
edge[cnt].to = v ;
head[u] = cnt++ ;
}
void dfs(int x ,int deep)
{
vis[x] = 1 ; // 标记访问过了
if(deep > maxs )
{
maxs = deep ;
maxi = x ;
}

for(int i = head[x] ; i!=-1 ; i= edge[i].next)
{

if(!vis[edge[i].to])
{
dfs(edge[i].to ,deep+1 ) ;
}
}



}
int main()
{
int n ;
scanf("%d",&n);
memset(head,-1,sizeof(head)) ;
for(int i = 1 ; i<=n-1 ; i++)
{
int u , v ;
scanf("%d%d",&u,&v);
Add(u,v) ;
Add(v,u) ;
}
maxs = -1 ;
maxi = -1 ;

dfs(1,1) ;
memset(vis,0,sizeof(vis)) ;
dfs(maxi,1) ;
printf("%d\n",maxs) ;

return 0;
}