编程语言
首页 > 编程语言> > 324 最近公共祖先 倍增算法

324 最近公共祖先 倍增算法

作者:互联网

视频链接:

// P3379 【模板】最近公共祖先(LCA)
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int N=5e5+10; 
int n,m,s,a,b;
vector<int> e[N];
int dep[N],fa[N][20];

void dfs(int u, int father){
  dep[u]=dep[father]+1;
  // 向上跳1,2,4...步的祖先节点
  fa[u][0]=father;
  for(int i=1; i<=19; i++) 
    fa[u][i]=fa[fa[u][i-1]][i-1]; 
    
  for(int v : e[u])
    if(v!=father) dfs(v, u);
}
int lca(int u, int v){
  if(dep[u]<dep[v])swap(u, v);
  // 先跳到同一层
  for(int i=19; i>=0; i--)
    if(dep[fa[u][i]]>=dep[v])
      u=fa[u][i];
  if(u==v) return v;
  // 再跳到LCA的下一层
  for(int i=19; i>=0; i--)
    if(fa[u][i]!=fa[v][i])
      u=fa[u][i], v=fa[v][i];
  return fa[u][0];
}
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);
  }
  dfs(s, 0);
  while(m--){
    scanf("%d%d", &a, &b);
    printf("%d\n",lca(a, b));
  }
  return 0;
}

 

标签:dep,include,int,d%,father,fa,算法,324,倍增
来源: https://www.cnblogs.com/dx123/p/16320461.html