【bzoj3124】[Sdoi2013]直径

1.求树的直径：

先随便取一个点，一遍dfs找到离它最远的点l1，再以l1为起点做一遍dfs，找到离l1最远的点l2
   那么l1到l2的距离即为直径
   
2. 求出有多少条边在这棵树的所有直径上：
   两个结论：
   1)这些边一定在同一条直径上
   2)它们一定是连续的 
   因此，只需要处理第一问求出的直径，从l2到l1求出每个点不经过这条直径所能到达的最远距离，若这个距离==它到l1/l2的距离，则这个点到l1/l2所经过的边都不在答案中，答案边的范围被缩小

 #include<algorithm>
 #include<iostream>
 #include<cstdlib>
 #include<cstring>
 #include<cstdio>
 #include<cmath>
 using namespace std;

 typedef long long LL;

 #define IN inline
 #define RG register

 #define N 400010

 struct edge
 {
     int to,next;
     LL w;
 }e[N<<];
 int head[N];
 int cnt;

 LL d[N],c[N],f[N],no[N];

 int u,v;
 LL w;

 int n;

 int ans;

 LL dis;

 bool flag;

 IN int getint()
 {
     int x=,f=;char ch=getchar();
     while(ch>''||ch<''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }

 IN int getLL()
 {
     LL x=,f=;char ch=getchar();
     while(ch>''||ch<''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }

 IN void link(int x,int y,LL z)
 {
     e[++cnt]=(edge){y,head[x],z};
     head[x]=cnt;
 }

 IN void dfs(int x,int fa,int & b)
 {
     f[x]=fa;
     if (dis<d[x])
     {
         dis=d[x];
         b=x;
     }
     for (RG int i=head[x];i;i=e[i].next)
     {
         int t=e[i].to;
         if (t!=fa)
         {
             d[t]=d[x]+e[i].w;
             dfs(t,x,b);
         }
     }
 }

 IN void work(int x,int fa)
 {
     if (dis<d[x])
         dis=d[x];
     for (RG int i=head[x];i;i=e[i].next)
     {
         int t=e[i].to;
         if (t!=fa && !no[t])
         {
             d[t]=d[x]+e[i].w;
             work(t,x);
         }
     }
 }

 int main()
 {
     int l1,l2;
     n=getint();
     for (RG int i=;i<n;i++)
     {
         u=getint();v=getint();w=getLL();
         link(u,v,w);
         link(v,u,w);
     }
     dfs(,,l1);
     dis=d[l1]=;
     dfs(l1,,l2);
     printf("%lld\n",dis);
     int l=l1,r=l2;
     for (RG int i=l2;i!=;i=f[i])
         no[i]=;
     for (RG int i=f[l2];i!=l1;i=f[i])
     {
         int r1=d[i],r2=d[l2]-d[i];
         dis=d[i]=;
         work(i,);
         if (dis==r1 && !flag)
             l=i,flag=true;
         if (dis==r2)
             r=i;
     }
     for (RG int i=r;i!=l;i=f[i])
         ans++;
     printf("%d",ans);
     return ;
 }