LCA模板

存一下模板啦。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cmath>
#include<queue>
using namespace std;
#define il inline
#define rg register
const int N=500020;
int i,j,k,v,p,L,q,e,R,tea,x,y,n,m,s,inde,maxn,cnt=1,maxlog=19;
int head[N],deep[N];
int f[500009][21];
struct Edge{
    int to,nxt;
}a[2*N];
int read(){                        //读入优化,可以照着这个模板来写,这个还算写的比较好看。 
    int re=0;
    char ch=getchar();
    while (ch<'0' || ch>'9') ch=getchar();
    while (ch>='0' && ch<='9'){ 
        re=re*10+ch-'0'; 
        ch=getchar();
    }
    return re;
}
il void add(int x,int y){
    inde++;
    a[inde].to=y;
    a[inde].nxt=head[x];
    head[x]=inde;
}
il int dfs(int x,int fa,int dep){//向下遍历的预处理 
    f[x][0]=fa;
    deep[x]=dep;
    for(rg int i=head[x];i;i=a[i].nxt){
        int y=a[i].to;
        if(y!=fa) dfs(y,x,dep+1);
    }
}
il void work(){
    for(int k=1;k<=maxlog;k++)
    for(rg int i=1;i<=n;i++){
        f[i][k]=f[f[i][k-1]][k-1];//玄学dp 
    }	
}
il int LCA(int x,int y){
     if(deep[y]>deep[x])swap(x,y);//deep[x]>deep[y]
     for(int i=maxlog;i>=0;i--){
         if(deep[y]<=deep[f[x][i]])
            x=f[x][i];
     } 
     if(x==y) return x;
     for(int i=maxlog;i>=0;i--){
        if(f[x][i]!=f[y][i]){
            x=f[x][i];
            y=f[y][i];
        }
     }
     return f[x][0];
}
int main(){
    cin>>n>>m>>s;//s是根节点,其他题目可能需要预处理 
    for(rg int i=1;i<n;i++){
        q=read();e=read();
        add(q,e);
        add(e,q);
    }
    dfs(s,0,1);
    work();
    for(rg int i=1;i<=m;i++){
        q=read();e=read();
        printf("%d\n",LCA(q,e));
    }
    return 0;
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值