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

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

储备知识

一、离线算法
离线算法其实就是将多个询问一次性解决.
二、深度优先生成树
什么是生成树：
在图论的数学领域中，如果连通图G的一个子图是一棵包含G的所有顶点的树，则该子图称为G的生成树(SpanningTree). 生成树是连通图的包含图中的所有顶点的极小连通子图. 图的生成树不惟一. 从不同的顶点出发进行遍历，可以得到不同的生成树.
由深度优先搜索得到的树为深度优先生成树.
------>无向图的DFS
三、并查集查询、合并最大祖先
四、最近公共祖先(Least Common Ancestors)
两个点在一棵无环棵树上距离最近的公共祖先节点.
LCA还可以将自己视为祖先节点.
五、链式前向星
什么是前向星：前向星是一种特殊的边集数组
利用前向星会有排序操作, 用快排时间至少为O(nlog(n))
为什么使用链式前向星：链式前向星可以避免排序
链式前向星建立边结构体:

struct Edge{
	int to;
	int next;
	int wei;
}edge[MAXN];

边结构体解释：
其中edge[i].to表示第i条边的终点, edge[i].next表示与第i条边同起点的下一条边在边集中的存储位置, edge[i].wei为边权值.
另外有一个数组head[MAXN],它是用来表示以i为起点的第一条边存在边集中的存储位置.
head[]数组一般初始化为-1.
加边连点的核心add函数:

void add(int x, int y){ //加边连点
    edge[++cnt].next = head[x];
     //记录与当前边同起点的下一条边在边集中的存储位置
    edge[cnt].to = y;
    //记录当前边的终点
    head[x] = cnt;
    //对当前边编号
}

注意事项

该模版默认为无向图Tarjan离线算法
若使用有向图Tarjan离线算法 则将add(y, x);以及que[y].push_back(make_pair(x, i));两行注释即可.

无向图测试题目

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

有向图测试数据

输入：
6 4 1
1 2
1 5
3 6
2 3
2 4
1 2
2 5
3 5
6 2
输出：
1
1
1
2

代码

#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]){  //如果当前子结点尚未访问
            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;
}