树的直径模板

树的直径模板

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