【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<<1];
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=0,f=1;char ch=getchar();
    while(ch>'9'||ch<'0'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
 
IN int getLL()
{
    LL x=0,f=1;char ch=getchar();
    while(ch>'9'||ch<'0'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';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=1;i<n;i++)
    {
        u=getint();v=getint();w=getLL();
        link(u,v,w);
        link(v,u,w);
    }
    dfs(1,0,l1);
    dis=d[l1]=0;
    dfs(l1,0,l2);
    printf("%lld
",dis);
    int l=l1,r=l2;
    for (RG int i=l2;i!=0;i=f[i])
        no[i]=1; 
    for (RG int i=f[l2];i!=l1;i=f[i])
    {
        int r1=d[i],r2=d[l2]-d[i];
        dis=d[i]=0;
        work(i,0);
        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 0;
}