树剖LCA

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
struct data{
    int next,to;
}sz[1000010];
struct Node{
    int fa,son,size,deep,top,num,end;
}tree[500010];
int head[500010];
int n,m,r,cnt,tot;
void update(int start,int end){
    sz[++cnt].next=head[start];
    sz[cnt].to=end;
    head[start]=cnt;
}
void dfs1(int u){
    tree[u].size=1;
    tree[u].son=0;
    for(int i=head[u];i;i=sz[i].next)
        if(sz[i].to!=tree[u].fa){
            tree[sz[i].to].fa=u;
            tree[sz[i].to].deep=tree[u].deep+1;
            dfs1(sz[i].to);
            tree[u].size+=tree[sz[i].to].size;
            if(tree[sz[i].to].size>tree[tree[u].son].size) tree[u].son=sz[i].to;
        }
    return;
}
void dfs2(int u,int tp){
    tree[u].top=tp;
    tree[u].num=++tot;
    if(tree[u].son){
        dfs2(tree[u].son,tp);
        for(int i=head[u];i;i=sz[i].next)
            if(sz[i].to!=tree[u].fa&&sz[i].to!=tree[u].son) dfs2(sz[i].to,sz[i].to);
    }
    tree[u].end=tot;
    return;
}
int lca(int a,int b){
    int f1=tree[a].top,f2=tree[b].top;
    while(f1!=f2){
        if(tree[f1].deep>tree[f2].deep){
            a=tree[f1].fa;
            f1=tree[a].top;
        }
        else{
            b=tree[f2].fa;
            f2=tree[b].top;
        }
    }
    if(tree[a].deep<tree[b].deep) return a;
    else return b;
}
int main(){
freopen("5.in","r",stdin);
freopen("5.out","w",stdout);
    scanf("%d%d%d",&n,&m,&r);
    int u,v;
    for(int i=1;i<=n;i++){
        scanf("%d%d",&u,&v);
        update(u,v);
        update(v,u);
    }
    dfs1(r);
    dfs2(r,r);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&u,&v);
        printf("%d\n",lca(u,v));
    }
    return 0;
}

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值