牛客网--桃花

链接：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;
}