树的直径模板

树的直径模板

http://codevs.cn/problem/1814/最长链.模板题
对于树上的最长链，有且仅有两种情况：

#include
#include
#include
#include
#include
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;   
}