#include <bits/stdc++.h>
using namespace std;
vector<int> arr[1000005];
int depth[500005];
int fa[500005][25];
int lg[1000005];
void dfs(int x,int dad)//建树求深度
{
fa[x][0]=dad,depth[x]=depth[dad]+1;
for(int i=1;i<=lg[depth[x]];i++) fa[x][i]=fa[fa[x][i-1]][i-1];
for(int i=0;i<arr[x].size();i++) if(dad!=arr[x][i]) dfs(arr[x][i],x);
}
int lca(int x,int y)
{
if(depth[x]<depth[y]) swap(x,y);
while(depth[x]>depth[y]) x=fa[x][lg[depth[x]-depth[y]]];
if(x==y) return x;
for(int i=lg[depth[x]];i>=0;i--)
{
if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
}
return fa[x][0];
}
int main()
{
int n,m,s;//节点数,询问次数,根节点
scanf("%d%d%d",&n,&m,&s);
for(int i = 1; i <= n; ++i)
lg[i] = lg[i-1] + (1 << lg[i-1] == i);
for(int i=1;i<=n;i++) lg[i]--;//计算log(x);
for(int i=1;i<n;i++)
{
int a,b;
scanf("%d%d",&a,&b);
arr[a].push_back(b);
arr[b].push_back(a);
}
dfs(s,0);
for(int i=0;i<m;i++)//求lcd
{
int a,b;
scanf("%d%d",&a,&b);
printf("%d\n",lca(a,b));
}
return 0;
}
LCA模板
于 2024-01-13 21:49:18 首次发布
本文介绍了如何在C++中使用深度优先搜索(DFS)和二分提升法计算最近公共祖先(LCA)来处理图的构建和查询操作,主要涉及数组和递归结构的使用。
摘要由CSDN通过智能技术生成