P-3379-LCA(倍增模板)

最新推荐文章于 2023-04-02 20:07:14 发布

Helium_wild

最新推荐文章于 2023-04-02 20:07:14 发布

阅读量167

点赞数

分类专栏：数据结构==LCA

本文链接：https://blog.csdn.net/qq_42576687/article/details/90111307

版权

数据结构==LCA 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目
 Reaper
在这里插入图片描述
输入样例#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

时空限制：1000ms,128M
数据：N<=500000，M<=500000

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+5,K=19;
struct Edge{int to,nex;}edge[N<<1];int head[N],tot;
void add(int from,int to){
    edge[++tot]=(Edge){to,head[from]},head[from]=tot;
    edge[++tot]=(Edge){from,head[to]},head[to]=tot;
}
int dep[N],fa[N][K+1];
void dfs(int x){
    for(int i=1;(1<<i)<=dep[x];++i) fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int i=head[x];i;i=edge[i].nex){
        int y=edge[i].to;
        if(y==fa[x][0]) continue;
        fa[y][0]=x,dep[y]=dep[x]+1,dfs(y);
    }
}
int lca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    int i;//上限i
    for(i=0;(1<<i)<=dep[x];++i);
    --i;
    //将x、y提到同一高度
    for(int j=K;j>=0;--j) if(dep[x]-(1<<j)>=dep[y]) x=fa[x][j];
    if(x==y) return x;
    //假如不找到上限i，而是j从K开始枚举 多组样例的话应该m(fa,-1)
    for(int j=K;j>=0;--j) if(fa[x][j]!=fa[y][j]) x=fa[x][j],y=fa[y][j];
    return fa[x][0];
}
int main(){
    int n,m,root;scanf("%d%d%d",&n,&m,&root);
    for(int i=1,x,y;i<n;++i) scanf("%d%d",&x,&y),add(x,y);
    dfs(root);
    while(m--){
        int x,y;scanf("%d%d",&x,&y);
        printf("%d\n",lca(x,y));
    }
}