dfs(tarjan)求lca

26 篇文章 0 订阅
4 篇文章 0 订阅

离线做法,可以做到o(n+m)
对于每一棵树,向下搜索,变成小树,如果子节点没有子节点了,就返回
然后在每一步中,找有x的查询,如果另一个访问过,就是他们的祖先
不同颜色的就是不同的子树
这里写图片描述

#include<cstdio>
#include<iostream>
using namespace std;
const int M=1001009;
int n,m,s,f[M],vis[M],lca[M*2];
int nex[M*2],head[M*2],to[M*2],tot,heade[M*2],nexe[M*2],toe[M*2],tote;
void add(int x,int y){
    nex[++tot]=head[x];
    to[tot]=y;
    head[x]=tot;
}
void adde(int x,int y){
    nexe[++tote]=heade[x];
    toe[tote]=y;
    heade[x]=tote;
}
int find(int x){
    return f[x]==x?x:f[x]=find(f[x]);
}
void dfs(int x){
    f[x]=x;//小树的根
    vis[x]=1;
    for(int i=head[x];i;i=nex[i]){
        int tmp=to[i];
        if(!vis[tmp]){
            dfs(tmp);//建小树
            f[tmp]=x;//把爸爸改回来
        }
    }
    for(int i=heade[x];i;i=nexe[i]){
        int tmp=toe[i];
        if(vis[tmp])
        {
            lca[i]=find(tmp);
            if(i%2) lca[i+1]=lca[i];//两个的祖先相同
            else lca[i-1]=lca[i];//
        }
    }
}
int main(){
    scanf("%d%d%d",&n,&m,&s);
    for(int i=1;i<n;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    for(int i=1;i<=m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        adde(x,y);
        adde(y,x);
    }
    dfs(s);
    for(int i=1;i<=m;i++)printf("%d\n",lca[i*2]);


}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值