BZOJ 3124: [Sdoi2013]直径 （随便写写就好了）

传送门
两次BFS（这个方法比较慢）—4492ms才跑完（~~(>_<)~~）
第一次bfs，随便找一个点作为根节点，找到树上到这个点距离最远的点，然后把他作为根节点跑第二次bfs，再找最远点，这就是直径
O(∩_∩)O哈哈~第一问成功解决
第二问略有些麻烦>o< >_<
树上直径的公共边一定是连续的直径的某一段边，这是显然的
首先我们第一次bfs可以得到若干个最远点，如果只有一个，那么就什么都不用做，如果不是一个怎么办呢?_?
找LCA（自已YY一下），找出所有最远点的LCA记为ansv1
第二次bfs也是一样，如果是一个什么都不用干，如果不是，找LCA记为ansv2
那么最终边数就是abs(dep[ansv1]-dep[ansv2])
AttentionO(∩_∩)O~~：
在两次bfs的时候，求LCA一定要更新dep数组和fa数组（因为root改变了）

---

代码如下：

#include<algorithm>#include<iostream>#include<cstring>#include<cstdio>#include<vector>#include<queue>#define LL long longusing namespace std;const int maxn=200000+5;int n,cnt,hd[maxn],to[maxn*2],nxt[maxn*2],ansv1,ansv2,ansv;LL dis[maxn],w[maxn*2];int fa[maxn][30+5],dep[maxn];void add(int x,int y,LL z){    to[cnt]=y;    w[cnt]=z;    nxt[cnt]=hd[x];    hd[x]=cnt++;}void init(){    for(int j=1;j<=30;j++)        for(int i=1;i<=n;i++)            fa[i][j]=fa[fa[i][j-1]][j-1];}int LCA(int a,int b){    if(dep[a]<dep[b])        swap(a,b);    int d=dep[a]-dep[b];    for(int i=0;i<=30;i++)        if((1<<i)&d)            a=fa[a][i];    if(a==b)        return b;    for(int i=30;i>=0;i--)        if(fa[a][i]!=fa[b][i])            a=fa[a][i],b=fa[b][i];    return fa[a][0];}int bfs(int root){    queue <int> q;    vector <int> point;    memset(dep,-1,sizeof(dep)),dep[root]=0,dis[root]=0,q.push(root);    while(!q.empty()){        int v=q.front();        q.pop();        for(int i=hd[v];i!=-1;i=nxt[i])            if(dis[to[i]]==-1)                dis[to[i]]=dis[v]+w[i],fa[to[i]][0]=v,dep[to[i]]=dep[v]+1,q.push(to[i]);    }    init();    int res=1;    for(int i=1;i<=n;i++)        res=dis[i]>dis[res] ? i : res;    ansv=res;    for(int i=1;i<=n;i++)        if(dis[i]==dis[res])            point.push_back(i);    if(point.size()==1)        return res;    res=point[0];    for(int i=1;i<point.size();i++){        res=LCA(res,point[i]);        if(res==root)            return res;    }    return res;}int main(){    memset(hd,-1,sizeof(hd)),cnt=0;    scanf("%d",&n);    LL z;    for(int i=1,x,y;i<n;i++)        scanf("%d%d%lld",&x,&y,&z),add(x,y,z),add(y,x,z);    memset(dis,-1,sizeof(dis)),ansv1=bfs(1);    memset(dis,-1,sizeof(dis)),ansv2=bfs(ansv);    cout<<dis[ansv]<<endl<<abs(dep[ansv2]-dep[ansv1])<<endl;    return 0;}

---

by >o< neighthorn