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

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

使用模版Tarjan离线算法求最近公共祖先(LCA)

代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <utility>
#include <cstring>
using namespace std;
#define inf 0x3f3f3f3f  //极大数据 10^9数量级
#define MAXN 500010 //点的个数 注意依题改变
vector<pair<int, int> > que[MAXN];  //存放查询关系
int vis[MAXN];  //标记访问数组
int pre[MAXN], head[MAXN];
//存放最大祖先 存放当前点为起点的第一条边在边集中的存储位置
int ans[MAXN];  //记录查询问题答案
int cnt, n, q, s;  //计数器 点的个数 问题查询数 数根节点
struct Edge{
    int to;
    int next;
}edge[MAXN<<1];    //边集
//edge[i].to表示第i条边的终点
//edge[i].next表示与第i条边同起点的下一条边在边集中的存储位置
void Init(){
    cnt = 0;    //计数器初始化为零
    memset(head, -1, sizeof(head)); //初始化head数组
    memset(vis, 0, sizeof(vis));    //初始化访问数组
    for (int i = 1; i <= n; i++){
        que[i].clear(); //初始化为无查询关系
        pre[i] = i; //初始化每个点都是自己的最大祖先
    }
}
void add(int x, int y){ //加边连点
    edge[++cnt].next = head[x]; //记录与当前边同起点的下一条边在边集中的存储位置
    edge[cnt].to = y;           //记录当前边的终点
    head[x] = cnt;              //对当前边编号
}
inline int find(int x){    //寻找最大祖先
    int r = x;
    while (pre[r] != r) r = pre[r];    //找到最大祖先
    int i = x, j;
    while (i != r){ //压缩路径 使每一后代直接祖先都为最大祖先
        j = pre[i];
        pre[i] = r;
        i = j;
    }
    return r;   //返回两者最大祖先
}
inline void join(int x, int y){    //合并关联群体 即连通分支
    int fx = find(x), fy = find(y);
    if (fx != fy) pre[fx] = fy; //最大祖先合并
}
void Tarjan(int x){ //查询的其中一个点
    vis[x] = 1; //标记已访问当前点
    for (int i = head[x]; i != -1; i = edge[i].next){
        int to = edge[i].to;    //得到当前边的终点
        //if (vis[to]) continue;
        if (!vis[to]){  //如果当前子结点尚未访问
            Tarjan(to); //查询子结点(当前边的终点)
            join(to, x);    //合并父子结点
        }
    }
    for (int i = 0; i < que[x].size(); i++){
        //que[x].size()为包含当前点的查询问题个数
        if (vis[que[x][i].first]) //如果查询问题中的第二个点也已访问标记过
            ans[que[x][i].second] = find(que[x][i].first);
        //则可记录两者最近共同祖先
    }
}
int main() {
    scanf("%d%d%d", &n, &q, &s);  //点的个数 问题查询数 数根节点
    Init(); //初始化
    int x, y;   //要查询的两个点
    for (int i = 1; i < n; i++){    //共有n-1条边
        scanf("%d%d", &x, &y);
        add(x, y);  //加边连点
        add(y, x);  //无向图 互相连接
    }
    for (int i = 0; i < q; i++){    //得到查询问题
        scanf("%d%d", &x, &y);
        que[x].push_back(make_pair(y, i));
        que[y].push_back(make_pair(x, i));    //查询问题中两个点互相绑定
    }
    Tarjan(s);  //算法实施
    for (int i = 0; i < q; i++) printf("%d\n", ans[i]); //输出答案
    return 0;
}