P3884 [JLOI2009]二叉树问题

---------------------

链接：Miku

---------------------

这一道题只需要在倍增lca的板子上改一改就可以了。

宽度和深度可以在倍增lca的dfs预处理的时候判断一下就可以，至于最后问的两点之间的距离

首先需要求出两点公共祖先的位置，然后计算他们深度的差，并且按照题目要求分别处理即可

--------------------

#include<iostream>
#include<cstdio>
using namespace std;
int head[*];
int p;
int md;
int mk;
int wei[];
struct b{
    int to;
    int ne;
} e[*];
int fa[*][];
int dep[*];
int n,m,s;
int logg[*];
int x,y;
void dfs(int now,int fat){
    dep[now]=dep[fat]+;
    md=max(md,dep[now]);
    wei[dep[now]]++;
    mk=max(mk,wei[dep[now]]);
    fa[now][]=fat;
    for(int i=;(<<i)<=dep[now];++i){
        fa[now][i]=fa[fa[now][i-]][i-];
    }
    for(int i=head[now];i;i=e[i].ne){
        if(e[i].to!=fat)
        dfs(e[i].to,now);
    }
}
int lca(int x,int y){
    int ans=;
    if(dep[x]>dep[y]){
        int v=dep[y];
    ans+=(*(dep[x]-dep[y]));
    while(dep[x]>dep[y]){
    x=fa[x][logg[dep[x]-dep[y]]-];
    }
    if(x==y)
    return ans;
    for(int k=logg[dep[x]]-;k>=;k--){
        if(fa[x][k]!=fa[y][k]){
            x=fa[x][k];
            y=fa[y][k];
        }
    }
    ans+=(*(v-dep[fa[x][]]));
    return ans;
    }
    else{
        ans+=(dep[y]-dep[x]);
        int v=dep[x];
    while(dep[x]<dep[y]){
    y=fa[y][logg[dep[y]-dep[x]]-];
    }
    if(x==y)
    return ans;
    for(int k=logg[dep[x]]-;k>=;k--){
        if(fa[x][k]!=fa[y][k]){
            x=fa[x][k];
            y=fa[y][k];
        }
    }
    ans+=(*(v-dep[fa[x][]]));
    return ans;

    }
}

void add(int f,int t){
    p++;
    e[p].to=t;
    e[p].ne=head[f];
    head[f]=p;
}
int main(){
    scanf("%d",&n);
    for(int i=;i<=n-;++i){
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs(,);
    cout<<md<<endl<<mk<<endl;
    for(int i=;i<=n;++i)
        logg[i]=logg[i-]+(<<logg[i-]==i);
    for(int i=;i<=;++i){
        scanf("%d%d",&x,&y);
        cout<<lca(x,y)<<endl;
    }
    return ;
}

Ac