算法描述
tarjan算法在求强联通分量效率十分高,核心思想是维护的dfn和low数组;而对于树上公共祖先问题也能离线地处理询问,算法复杂度 O ( n + q ) O(n+q) O(n+q)。算法是:后序遍历树上的结点,每次的子孙结点的祖先集合合并到其父亲结点中(并查集),之后去查询当前结点是否有询问,有询问且两个点都已经访问过那么这个两个点的LCA是另外那个点的祖先。由于是后序遍历的缘故,当前结点和另外那个结点假设同处与一个询问中,那个结点如果已经被访问,那么它的时间戳势必比当前的低,说明那颗搜索树已经完全访问完毕。那么当前结点访问的时候那个结点肯定已经更新到和当前结点最早连接的地方了,因为本结点所在的搜索树就是从那边过来的。
例子
以HDU2586为例,除了要LCA还要预处理一下树上差分,方便计算路径和。
#include <bits/stdc++.h>
using namespace std;
/**
* HDU2586
* tarjan求LCA 并查集
* 问题:
* n点n-1边的无向图求询问的两点的最短路
* 分析:
* 显然这n个点可以构成一个树(任意两点有且仅有一条路),任取一个点作为根求该树的最近公共祖先即可
* 当然首先要取处理每个点的深度
* answer = presum[u] + presum[v] - 2*presum[lca(T,u,v)]
*/
struct Edge {
int to;
int w;
int next;
};
Edge E[40005];
int head[40005], cnt;
int fa[40005]; //并查集
int presum[40005]; //树上前缀和
int n, m;
int qry[205][2], ans[205];
void edge_add(int u, int v, int w) {
E[cnt].to = v;
E[cnt].w = w;
E[cnt].next = head[u];
head[u] = cnt++;
}
void init() {
memset(head, -1, sizeof(head));
memset(fa, -1, sizeof(fa));
memset(presum, -1, sizeof(presum));
memset(ans, -1, sizeof(ans));
cnt = 0;
int u, v, w;
for (int i = 1; i < n; i++) {
scanf("%d%d%d", &u, &v, &w);
edge_add(u, v, w);
edge_add(v, u, w);
}
for (int i = 0; i < m; i++)
scanf("%d%d", &qry[i][0], &qry[i][1]);
}
void dfs(int rt, int sum) {
presum[rt] = sum; //记录从root到当前结点的路径权值和
for (int cur = head[rt]; ~cur; cur = E[cur].next)
if (presum[E[cur].to] == -1)
dfs(E[cur].to, sum+E[cur].w);
}
int find(int rt) {
while (fa[rt]^rt) rt = fa[rt] = fa[fa[rt]]; //不完全的路径压缩
return rt;
}
void merge(int u, int v) {
int a=find(u);
int b=find(v);
if (a^b) fa[b] = a;
}
void tarjan(int rt) {
fa[rt] = rt;
for (int cur = head[rt]; ~cur; cur = E[cur].next)
if (fa[E[cur].to] == -1) {
tarjan(E[cur].to);
merge(rt, E[cur].to); //合并到根结点
}
for (int i = 0; i < m; i++) {
if (ans[i]^-1) continue; //已解决
if (qry[i][0] == rt && fa[qry[i][1]]^-1 || qry[i][1] == rt && fa[qry[i][0]]^-1) {
int v = qry[i][0]^qry[i][1]^rt;
int k = find(v); //找到LCA(T, rt, v)
ans[i] = presum[rt]+presum[v]-presum[k]-presum[k];
}
}
}
int main() {
int T;
scanf("%d", &T);
while (T--) {
scanf("%d%d", &n, &m);
init();
dfs(1, 0);
tarjan(1);
for (int i = 0; i < m; i++)
printf("%d\n", ans[i]);
}
return 0;
}