Java教程

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;
}

 

这篇关于325 最近公共祖先 Tarjan算法的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!