树上最近公共祖先(Tarjan+并查集)

25 篇文章 0 订阅

算法描述

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;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小胡同的诗

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值