倍增求LCA模板

#include<bits/stdc++.h>
using namespace std;

typedef long long LL;
const int maxn=5e5+10,maxk=1e4+10,lenOfNode=52274,INF=0x3f3f3f3f;
int lg[maxn],n,m,d[maxn],f[maxn][21],dep[maxn],cnt;
int nxt[2*maxn],to[2*maxn],w[2*maxn],he[2*maxn];
//f[x][i]为x往上走2的i次方步的祖先,lg[x]为x取2的对数,dep[x]为x的深度,d[x]为x到根节点的距离
void add(int x,int y,int dis){ to[++cnt]=y,w[cnt]=dis,nxt[cnt]=he[x],he[x]=cnt; }
inline int read(){ //加快输入
    char ch;
    while ((ch=getchar())<'0'||ch>'9') continue;
    int ret=0;
    while (ch>='0'&&ch<='9') ret=(ret<<1)+(ret<<3)+ch-'0',ch=getchar();
    return ret;
}

void dfs(int u,int father){ //nlogn预处理出f、dep和d数组
    dep[u]=dep[father]+1;
    for (int i=0;i<20;i++) f[u][i+1]=f[f[u][i]][i];
    for (int e=he[u];e;e=nxt[e]){
        int v=to[e];
        if (v==father) continue;
        f[v][0]=u,d[v]=d[u]+1;
        dfs(v,u);
    }
}
int LCA(int x,int y){ //倍增求LCA
    if (dep[x]<dep[y]) swap(x,y);
    while (dep[x]>dep[y]) x=f[x][lg[dep[x]-dep[y]]];  //跳到同一层,用lg可以减少跳跃时间
    if (x==y) return x;
    for (int i=lg[dep[x]];i>=0;i--) if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i]; //一起往上跳
    return f[x][0];
}
inline int path(int x,int y,int &anc){ anc=LCA(x,y); return d[x]+d[y]-2*d[anc]; }  //求x到y两个节点间的路径长度
int main(){
    scanf("%d%d",&n,&m);
    for (int i=0;i<n-1;i++){
        int x,y;
        x=read(),y=read();
        add(x,y,1),add(y,x,1);
    }
    f[1][0]=0,dep[0]=-1,d[1]=0;
    dfs(1,0);
    lg[0]=-1; for (int i=1;i<=n;i++) lg[i]=lg[i>>1]+1;

    for (int i=0;i<m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d\n",LCA(x,y));
    }
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值