1. 程式人生 > 其它 >325 最近公共祖先 Tarjan演算法

325 最近公共祖先 Tarjan演算法

視訊連結:

// P3379 【模板】最近公共祖先(LCA)
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;

const int N=500005,M=2*N;
int n,m,s,a,b;
vector<int> e[N];
vector<pair<int,int>>query[N];
int fa[N],vis[N],ans[M]; 

int find(int x){
  
if(x==fa[x]) return x; return fa[x]=find(fa[x]); } void tarjan(int x){ vis[x]=true;//標記x已訪問 for(auto y : e[x]){ if(!vis[y]){ tarjan(y); fa[y]=x;//回到x時指向x } } //離開x時找LCA for(auto q : query[x]){ int y=q.first,i=q.second; if(vis[y])ans[i]=find(y); } } int main(){ scanf(
"%d%d%d", &n,&m,&s); for(int i=1; i<n; i++){ scanf("%d%d",&a,&b); e[a].push_back(b); e[b].push_back(a); } for(int i=1;i<=m;i++){ scanf("%d%d",&a,&b); query[a].push_back({b,i}); query[b].push_back({a,i}); } for(int i=1;i<=N;i++)fa[i]=i; tarjan(s);
for(int i=1; i<=m; i++) printf("%d\n",ans[i]); return 0; }