树上最近公共祖先(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
    评论
Tarjan算法是一种用于求解最近公共祖(Least Common Ancestors,LCA)问题的离线算法。算法的核心思想是利用深度优先搜索(DFS)和并查集(Union Find)来解决问题。 首先,我们从根节点开始遍历每一个节点,并将节点分为三类,用st[]数组表示。0代表还未被遍历,1代表正在遍历这个点,2代表已经遍历完这个点并且回溯回来了。这样的划分有助于确定节点的最近公共祖先。 在Tarjan算法中,我们一边遍历一边回应查询。每当遍历到一个节点时,我们查找与该节点相关的所有查询。如果查询中的节点已经被遍历完(即st[]值为2),我们可以利用已经计算好的信息来计算它们的最近公共祖先最近公共祖先的距离可以通过两个节点到根节点的距离之和减去最近公共祖先节点到根节点的距离来计算。 在Tarjan算法中,我们可以通过深度优先搜索来计算dist[]数组,该数组表示每个节点到根节点的距离。我们可以利用父节点到根节点的距离加上边的权值来计算每个节点到根节点的距离。 最后,我们可以通过并查集来操作st[]数组。当遍历完一个节点的所有子树后,将子树中的节点放入该节点所在的集合。这样,每个子树的节点的最近公共祖先都是该节点。 综上所述,Tarjan算法利用DFS和并查集来求解最近公共祖先问题。它的时间复杂度为O(n+m),其中n是节点数,m是查询次数。通过该算法,我们可以高效地解决最近公共祖先的问题。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [最近公共祖先tarjan](https://blog.csdn.net/qq_63092029/article/details/127737575)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"] - *2* [【模版】Tarjan离线算法求最近公共祖先(LCA)](https://blog.csdn.net/weixin_43359312/article/details/100823178)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"] - *3* [Tarjan算法求解最近公共祖先问题](https://blog.csdn.net/Yeluorag/article/details/48223375)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

小胡同的诗

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

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

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

打赏作者

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

抵扣说明:

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

余额充值