【模板】最近公共祖先(LCA)
题目描述
如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先。
输入格式
第一行包含三个正整数 N , M , S N,M,S N,M,S,分别表示树的结点个数、询问的个数和树根结点的序号。
接下来 N − 1 N-1 N−1 行每行包含两个正整数 x , y x, y x,y,表示 x x x 结点和 y y y 结点之间有一条直接连接的边(数据保证可以构成树)。
接下来 M M M 行每行包含两个正整数 a , b a, b a,b,表示询问 a a a 结点和 b b b 结点的最近公共祖先。
输出格式
输出包含 M M M 行,每行包含一个正整数,依次为每一个询问的结果。
样例 #1
样例输入 #1
5 5 4
3 1
2 4
5 1
1 4
2 4
3 2
3 5
1 2
4 5
样例输出 #1
4
4
1
4
4
提示
对于 30 % 30\% 30% 的数据, N ≤ 10 N\leq 10 N≤10, M ≤ 10 M\leq 10 M≤10。
对于 70 % 70\% 70% 的数据, N ≤ 10000 N\leq 10000 N≤10000, M ≤ 10000 M\leq 10000 M≤10000。
对于 100 % 100\% 100% 的数据, N ≤ 500000 N\leq 500000 N≤500000, M ≤ 500000 M\leq 500000 M≤500000。
样例说明:
该树结构如下:
第一次询问: 2 , 4 2, 4 2,4 的最近公共祖先,故为 4 4 4。
第二次询问: 3 , 2 3, 2 3,2 的最近公共祖先,故为 4 4 4。
第三次询问: 3 , 5 3, 5 3,5 的最近公共祖先,故为 1 1 1。
第四次询问: 1 , 2 1, 2 1,2 的最近公共祖先,故为 4 4 4。
第五次询问: 4 , 5 4, 5 4,5 的最近公共祖先,故为 4 4 4。
故输出依次为 4 , 4 , 1 , 4 , 4 4, 4, 1, 4, 4 4,4,1,4,4。
2021/10/4 数据更新 @fstqwq:应要求加了两组数据卡掉了暴力跳。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define de(x) cout<<x<<" ";
#define sf(x) scanf("%d",&x);
#define Pu puts("");
const int N=5e5+10;
int hed[N],nxt[N<<1],var[N<<1],tot;
int dep[N],f[N][20];
int n,m,s,lg;
void _add(int x,int y){
var[++tot]=y;
nxt[tot]=hed[x];
hed[x]=tot;
}
void dfs(int u,int v){//第一种:dfs 来完善f[i][j]
dep[v]=dep[u]+1;
f[v][0]=u;//玄学1:把每个点的父亲都确定
for(int i=1;i<=lg;i++){//注意这一步不能省略
f[v][i]=f[f[v][i-1]][i-1];
}
for(int i=hed[v];i;i=nxt[i]){
int t=var[i];
if(t!=u){
dfs(v,t);
}
}
}
void dfs1(int x){//第二种:dfs+dp 来完善f[i][j]
for(int i=hed[x];i;i=nxt[i]){
int t=var[i];
if(t!=f[x][0]){
dep[t]=dep[x]+1;
f[t][0]=x;
dfs1(t);
}
}
}
int lca(int x,int y){
if(dep[x]>dep[y]) swap(x,y);
for(int i=lg;i>=0;i--){
if(dep[f[y][i]]>=dep[x]){
y=f[y][i];//调整到同一深度
}
}
if(x==y) return x;//防止是父子关系
for(int i=lg;i>=0;i--){
if(f[x][i]!=f[y][i]){
x=f[x][i];y=f[y][i];
}//向上找它们的最小父亲
}
return f[x][0];
}
int main(){
cin>>n>>m>>s;
lg=log(n)/log(2)+1;
int x,y;
for(int i=1;i<=n-1;i++){
sf(x)sf(y)
_add(x,y);_add(y,x);
}
//dfs(0,s);
f[s][0]=s;//也可以这样写,注意这里需要把s的父亲定义为它自己
dfs1(s);
for(int j=1;j<=lg;j++){//玄学2:用dp把所有父亲数组f[i][j],i向上第(2^j)个父亲确定
for(int i=1;i<=n;i++){
f[i][j]=f[f[i][j-1]][j-1];
}
}
for(int i=1;i<=m;i++){
sf(x)sf(y)
printf("%d\n",lca(x,y));
}
return 0;
}