LCA板子 求调
查看原帖
LCA板子 求调
310054
WITCHER2077楼主2023/2/3 16:07
#include<bits/stdc++.h>
using namespace std;
const int maxn=500005;
int h[maxn],cnt;
struct edge{
    int to,next;
}a[maxn*2];
void add(int x,int y){
    a[++cnt].next=h[x];
    a[cnt].to=y;
    h[x]=cnt;
}
int n,m,s,de[maxn],fa[maxn][30],ln;
void dfs(int x,int f){
    fa[x][0]=f;
    de[x]=de[f]+1;
    for(int j=1;j<=ln;j++){
        if((1<<j)>=de[x]) break;
        fa[x][j]=fa[fa[x][j-1]][j-1];
    }
    for(int i=h[x];i;i=a[i].next){ 
        int y=a[i].to;
        if(y==f) continue;
        dfs(y,x);
    }
}
int lca(int x,int y){
    if(de[x]<de[y]) swap(x,y);
    int cd=de[x]-de[y];
    for(int i=ln;i>=0;i--){
//      if(de[x]==de[y]) break;
        if((1<<i)&cd) x=fa[x][i];
    }
    if(x==y) return x;
    for(int i=ln;i;i--){
        if(fa[x][i]==fa[y][i]) continue;
        x=fa[x][i];
        y=fa[y][i];
    }
    return fa[x][0];
}
int main(){
//  freopen("1.in","r",stdin);
//  freopen("out.txt","w",stdout);
    cin>>n>>m>>s;
    ln=log(n)/log(2)+1;
    for(int i=1;i<n;i++){
        int x,y;
        cin>>x>>y;
        add(x,y);
        add(y,x);
    }
    dfs(s,0);
    while(m--){
        int x,y;
        cin>>x>>y;
        cout<<lca(x,y)<<endl;
    }
    return 0;
}
2023/2/3 16:07
加载中...