C++ P3379 【模板】最近公共祖先(LCA)

 题目地址:https://www.luogu.org/problemnew/show/P3379


主要是用来作为参考代码的。

#include<cstdio>
#include<iostream>
using namespace std;
int cnt=0,head[1000010],f[500010][21],d[1000010];
struct Edge{
    int v,nxt;
}e[1000010];//边 
void addEdge(int u,int v){
    e[++cnt].v=v;
    e[cnt].nxt=head[u];
    head[u]=cnt;
}//建图(加边) 
void dfs(int u,int fa){
    d[u]=d[fa]+1;//初始化点的深度 
    f[u][0]=fa;//结点u的往上第2^0(也就是1)为自己的直接父结点,
    //因此直接设置,并为之后的计算做准备 
    for(int i=1;(1<<i)<=d[u];i++){//循环中i的判断是否可以继续循环状态:2^1 2^2 2^3 
        //也就是从下往上跑深度 
        f[u][i]=f[f[u][i-1]][i-1];
        //乘法原理可得:2^(i-1)+2^(i-1)=2^(i-1)*2=2^i 
    }
    for(int i=head[u];i;i=e[i].nxt){
        if(e[i].v!=fa)dfs(e[i].v,u);
    }//BFS 如果下一个点不是自己的父节点,就继续bfs 
}
int lca(int a,int b){
    if(d[a]>d[b])swap(a,b);
    //先保证a的深度小于b的深度,因为可能要将其中一个较深点的位置暂时提到上面 
    //也就是保证b在下面,因为我们判断的是深度
    for(int i=20;i>=0;i--){
        if(d[a]<=d[b]-(1<<i))b=f[b][i];
        //b向上跳(因为我们保证了b在下面),一直到a与b深度相同为止.
        //某不科学的32位int数b:jump jump~
    } 
    if(a==b)return a;
    //如果a和b序号相同,也就是在同一个点上
    //就可以返回a(或b,因为实际是一个点)了 
    for(int i=20;i>=0;i--){
        if(f[a][i]==f[b][i])continue;//同上。如果超了或到了就continue(反正就20次操作) 
        a=f[a][i];b=f[b][i];
    } 
    return f[a][0];
}
int main(){
    int n,m,s;
    scanf("%d%d%d",&n,&m,&s);
    for(int i=1;i<n;i++){
    	int ta,tb;
        scanf("%d%d",&ta,&tb);
        addEdge(ta,tb);
        addEdge(tb,ta);
    }//普通的加边建图 
    dfs(s,0);
    for(int i=1;i<=m;i++){
    	int ta,tb;
        scanf("%d%d",&tb,&ta);
        printf("%d\n",lca(ta,tb));
    }
    return 0;
} 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值