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

最新推荐文章于 2025-10-11 20:08:20 发布

原创最新推荐文章于 2025-10-11 20:08:20 发布 · 800 阅读

0 ·

CC 4.0 BY-SA版权

BZOJ 专栏收录该内容

45 篇文章

订阅专栏

传送门
两次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 long
using 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;
}