P3379 【模板】最近公共祖先（LCA）-dfs+dp

最新推荐文章于 2024-08-22 17:27:46 发布

JYWsuperman

最新推荐文章于 2024-08-22 17:27:46 发布

阅读量91

点赞数

分类专栏： # 洛谷题单文章标签：深度优先算法图论

本文链接：https://blog.csdn.net/weixin_52490191/article/details/125887449

版权

洛谷题单专栏收录该内容

75 篇文章 1 订阅

订阅专栏

【模板】最近公共祖先（LCA）

题目描述

如题，给定一棵有根多叉树，请求出指定两个点直接最近的公共祖先。

输入格式

第一行包含三个正整数 $N, M, S$ ，分别表示树的结点个数、询问的个数和树根结点的序号。

接下来 $N - 1$ 行每行包含两个正整数 $x, y$ ，表示 $x$ 结点和 $y$ 结点之间有一条直接连接的边（数据保证可以构成树）。

接下来 $M$ 行每行包含两个正整数 $a, b$ ，表示询问 $a$ 结点和 $b$ 结点的最近公共祖先。

输出格式

输出包含 $M$ 行，每行包含一个正整数，依次为每一个询问的结果。

样例 #1

样例输入 #1

样例输出 #1

提示

对于 $30\%$ 的数据， $N\leq 10$ ， $M\leq 10$ 。

对于 $70\%$ 的数据， $N\leq 10000$ ， $M\leq 10000$ 。

对于 $100\%$ 的数据， $N\leq 500000$ ， $M\leq 500000$ 。

样例说明：

该树结构如下：

第一次询问： $2, 4$ 的最近公共祖先，故为 $4$ 。

第二次询问： $3, 2$ 的最近公共祖先，故为 $4$ 。

第三次询问： $3, 5$ 的最近公共祖先，故为 $1$ 。

第四次询问： $1, 2$ 的最近公共祖先，故为 $4$ 。

第五次询问： $4, 5$ 的最近公共祖先，故为 $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;
}