牛客-牛客小白月赛6-C 桃花

                                                C-桃花

链接：https://www.nowcoder.com/acm/contest/136/C
来源：牛客网

题目描述 
    桃花一簇开无主，可爱深红映浅红。

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

输入描述:
第一行有一个正整数n，表示桃树的节点个数。
接下来n-1行，第i行两个正整数ai,bi ，表示桃树上的节点ai,bi之间有一条边。
输出描述:
第一行一个整数，表示HtBest使用一次魔法棒最多可以摘到多少桃花。

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

 

思路：题目要求找出树中最长的一条线段Len，可以用搜索来查找，从任意一点s开始搜索，遍历s的所有连接点，找出最大长度的线段，那么最长的线段Len是包含s的最长线段 L1 和不包含s的最长线段 L2 ，而 L1就是搜索的最大长度，而 L2 则是 当前点搜索的最长两条边的长度和+1。

Code :

#include<iostream>
#include<vector>
#include<cstring>
using namespace std;

struct E{
    int to,next;
};
const int MAX_N=1000005;
int n,ans,iid;
int head[MAX_N];
E edge[MAX_N<<1];
vector<int> e[MAX_N];
bool boo[MAX_N];

inline void add_edge(int u,int v)
{
    edge[iid].to=v;
    edge[iid].next=head[u];
    head[u]=iid++;
}
int DFS(int k,int pre);
int main()
{
	memset(head,-1,sizeof(head));
	iid=0;
	scanf("%d",&n);
	int u,v;
	for(int i=1;i<n;++i)
	{
		scanf("%d%d",&u,&v);
		e[u].push_back(v);
		e[v].push_back(u);
		add_edge(u,v);
        add_edge(v,u);
	}
	ans=1;
	for(int i=1;i<=n;++i)
		if(boo[i]==false){
			boo[i]=true;
			ans=max(ans,DFS(i,0));
		}
	printf("%d\n",ans);
	
	return 0;
}

int DFS(int k,int pre)
{
	int sum1=0,sum2=0,sum=0;
	for(int i=head[k];i!=-1;i=edge[i].next)
	{
        int c=edge[i].to;
		if(c!=pre&&boo[c]==false){
			boo[c]=true;
			sum=DFS(c,k);
			if(sum>=sum1){
				sum2=sum1;	sum1=sum;
			}else	if(sum>sum2)	sum2=sum;
		}
	}
	ans=max(ans,sum1+sum2+1);
	return sum1+1;
}