P3379 【模板】最近公共祖先（LCA）——Tarjan做法

//https://www.luogu.com.cn/problem/P3379
#include <bits/stdc++.h>
using namespace std;
const int N=5e5+10,Q=5e5+10,M=2*N;
typedef pair<int,int> pii;
int e[M],ne[M],h[N],idx;
int p[N];
int res[Q];
int st[N];
int n,m,root;
vector<pii> query[N];
void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int find(int x){
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
void tarjan(int u){
    st[u]=1;
    for(int i=h[u];~i;i=ne[i]){
        int j=e[i];
        if(!st[j]){
            tarjan(j);
            p[j]=u;
        }
    }
    for(auto item:query[u]){
        int y=item.first,id=item.second;
        if(st[y]==2){
            res[id]=find(y);
        }
    }
    st[u]=2;
}
int main(){
    scanf("%d%d%d",&n,&m,&root);
    memset(h,-1,sizeof h);
    for(int i=1;i<=n;i++) p[i]=i;
    for(int i=0,a,b;i<n-1;i++){
        scanf("%d%d",&a,&b);
        add(a,b),add(b,a);
    }
    for(int i=0,a,b;i<m;i++){
        scanf("%d%d",&a,&b);
        if(a==b){
			res[i]=a;
			continue;
		}
        query[a].push_back({b,i});
        query[b].push_back({a,i});
    }
    tarjan(root);
    for(int i=0;i<m;i++) printf("%d\n",res[i]);
    return 0;
}