树的直径模板
http://codevs.cn/problem/1814/最长链.模板题
对于树上的最长链,有且仅有两种情况:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
int n,tot,a,b,ans,pos;
int first[200010],nxt[200010];
struct edge
{
int u,v;
}l[200010];
void build(int f,int t)
{
l[++tot]=(edge){f,t};
nxt[tot]=first[f];
first[f]=tot;
}
void dfs(int k,int fa,int d)
{
if(d>ans)
{
ans=d;
pos=k;
}
for(int i=first[k];i!=-1;i=nxt[i])
{
int x=l[i].v;
if(x==fa)
continue;
dfs(x,k,d+1);
}
}
int main()
{
memset(first,-1,sizeof(first));
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&a,&b);
if(a!=0)
{
build(i,a);
build(a,i);
}
if(b!=0)
{
build(i,b);
build(b,i);
}
}
dfs(1,1,0);//第一遍DFS找出树上深度最深的点
ans=0;
dfs(pos,1,0);//第二遍DFS由深度最深的点向上找最长链
printf("%d",ans);
return 0;
}