洛谷 – P3379 -【模板】最近公共祖先(树链剖分)

https://www.luogu.org/problem/P3379
这次我是用树剖写的LCA,写的比倍增简单而且还快一点。
树剖一般指的是重链剖分,首先把所有节点的子节点中最大的标出来,称为重子节点,从节点到重子节点的边叫重边,一段连续的重边称为重链,其他的轻子节点叫轻链,然后把整棵树dfs一下,dfs的时候优先dfs重儿子,这样可以保证同一条重链上的点时间戳连续。这样就把一棵树剖分成了很多条链,那么就可以用数据结构维护这些链,优化时间复杂度。
如果询问两个点的LCA,先判断是不是在一条链上,如果不是那么要先比较这两个点的链头的深度,把深度大的点移到链头的父节点上,直到这两个点在一条链上,然后输出深度较小的点就好了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 5e5 + 10, mod = 1e9 + 7;

vector<int> g[maxn];
int fa[maxn],dep[maxn],sz[maxn],son[maxn],top[maxn];
//父亲、深度、大小、儿子、时间戳、重链顶、时间戳对应的编号、时间
void dfs1(int u, int p, int d)
{
    fa[u] = p; dep[u] = d; sz[u] = 1;
    int maxson = 0;
    for(int v : g[u]){
        if (v != p){
            dfs1(v, u, d + 1);
            sz[u] += sz[v];
            if(sz[v] > maxson){
                maxson = sz[v];
                son[u] = v;
            }
        }
    }
}
void dfs2(int u,int topf)
{
    top[u] = topf;
//    tid[u] = ++cnt;
//    rnk[cnt] = u;
    if (son[u]){
        dfs2(son[u], topf);
        for(int v : g[u]){
            if(v != fa[u] && v != son[u]) dfs2(v, v);
        }
    }
}

int main()
{
    int n, m, u, v, s;
    scanf("%d %d %d", &n, &m, &s);
    int t = n;
    while (--t){
        scanf("%d %d", &u, &v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs1(s, s, 1);
    dfs2(s, s);
    while (m--){
        scanf("%d %d", &u, &v);
        while (top[u] != top[v]){
            if (dep[top[u]] > dep[top[v]]) u = fa[top[u]];
            else v = fa[top[v]];
        }
        printf("%d\n", dep[u] < dep[v] ? u : v);
    }
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值