1. 程式人生 > >luogu_3379 【模板】最近公共祖先(LCA)

luogu_3379 【模板】最近公共祖先(LCA)

span oid ont return mes ace print next using

#include<bits/stdc++.h>

using namespace std;

#define N 500010*2

struct edge{int v,next;}e[N];

struct qedge{int u,v,next,lca,num;}e1[N];

int n,m,t,head[N],top,top1,ans[N],head1[N],p[N];

bool vis[N];

int find(int x){return x==p[x]?x:p[x]=find(p[x]);}

void combine(int a,int b){p[find(b)]=find(a);}

void add(int u,int v){

e[++top].v=v; e[top].next=head[u]; head[u]=top;

e[++top].v=u; e[top].next=head[v]; head[v]=top;

}

void qadd(int u,int v,int x){

e1[++top1].v=v; e1[top1].next=head1[u]; e1[top1].num=x; head1[u]=top1;

e1[++top1].v=u; e1[top1].next=head1[v]; e1[top1].num=x; head1[v]=top1;

}

void tarjin_lca(int u){

vis[u]=1; p[u]=u;

for(int i=head[u];i;i=e[i].next){

int v=e[i].v;

if(!vis[v]){

tarjin_lca(v);

combine(u,v);

}

}

for(int i=head1[u];i;i=e1[i].next){

int v=e1[i].v;

if(vis[v]){

ans[e1[i].num]=find(v);

}

}

}

int main(){

scanf("%d%d%d",&n,&m,&t);

for(int i=1;i<n;i++){

int u,v;

scanf("%d%d",&u,&v);

add(u,v);

}

for(int i=1;i<=m;i++){

int u,v;

scanf("%d%d",&u,&v);

qadd(u,v,i);

}

tarjin_lca(t);

for(int i=1;i<=m;i++)printf("%d\n",ans[i]);

return 0;

}

luogu_3379 【模板】最近公共祖先(LCA)