题目描述
如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先。
输入格式
第一行包含三个正整数N、M、S,分别表示树的结点个数、询问的个数和树根结点的序号。
接下来N-1行每行包含两个正整数x、y,表示x结点和y结点之间有一条直接连接的边(数据保证可以构成树)。
接下来M行每行包含两个正整数a、b,表示询问a结点和b结点的最近公共祖先。
输出格式
输出包含M行,每行包含一个正整数,依次为每一个询问的结果。
基于倍增思想求LCA步骤:
1.先预处理出各节点的深度和他们2^j的级的祖先
2.预处理出log2(i)+1的值
3.计算lca
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
typedef long long ll;
const int maxn=5e5+7;
const int inf=1e7+7;
int n,m,s;
vector<int> g[maxn];
int dep[maxn],f[maxn][22],lg[maxn];
//f[i][j]表示i往上走2^j步到达的点
void dfs(int u,int fa){
dep[u]=dep[fa]+1;
f[u][0]=fa;
for(int j=1;(1<<j)<=dep[u];j++)//u的2^(j-1)祖先的2^(j-1)祖先
f[u][j]=f[f[u][j-1]][j-1];
for(int i=0;i<(int)g[u].size();i++){
int v=g[u][i];
if(v!=fa)dfs(v,u);
}
}
void init(){
for(int i=1;i<=n;i++)//预处理出log2(i)+1的值,便于直接调用
lg[i]=lg[i-1]+(1<<lg[i-1]==i);
}
int lca(int u,int v){
if(dep[u]<dep[v])swap(u,v);//不妨设u的深度 >= v的深度,便于下面统一处理
while(dep[u]>dep[v])u=f[u][lg[dep[u]-dep[v]]-1];//跳到统一深度
if(u==v)return u;
for(int j=lg[dep[u]]-1;j>=0;j--)//从步数大往小的跳
if(f[u][j]!=f[v][j])//因为我们要跳到它们LCA的下面一层,所以它们肯定不相等,如果不相等就跳过去
u=f[u][j],v=f[v][j];
return f[u][0];//返回父节点
}
int main(){
scanf("%d%d%d",&n,&m,&s);
for(int i=0;i<n-1;i++){
int u,v;scanf("%d%d",&u,&v);
g[u].push_back(v);
g[v].push_back(u);
}
dfs(s,0);
init();
for(int i=0;i<m;i++){
int u,v;scanf("%d%d",&u,&v);
printf("%d\n",lca(u,v));
}
return 0;
}