蓝桥杯铺垫:树的应用之树链剖分

b站的讲解视频:https://www.bilibili.com/video/BV1tY4y1G7em/?spm_id_from=333.788.recommend_more_video.0&vd_source=fd7c48bc1fba0b13fb26b253475a99ca

模板题目:https://www.luogu.com.cn/problem/P3379

首先我们要明白一些名词:

重儿子:父结点的所有儿子中子树结点最多的结点
轻儿子:父结点中除重儿子以外的儿子
重边:父节点和重儿子连的边
轻边:父节点和轻儿子连的边
重链:由多条重边连成的边

性质:(建议自己画课树体会一下)

1.整棵树会剖分为若干重链
2.轻儿子是每条重链顶点
3.任意一条路径可以切分出的链不超过log n条代码:

#include<bits/stdc++.h>

using namespace std;

const int N = 5e5 + 5;

vector<int>e[N];//存边,如e[2]==1,则第2,第1结点之间有边

int  fa[N] = { };//fa[i]为i的父节点
int dep[N] = { };//dep[i]为i的深度
int son[N] = { };//son[i]为i重儿子
int  sz[N] = { };//sz[i]为以i为根的字数节点数
int top[N] = { };//top[i]为i所在重链顶点

void dfs1(int u, int father) {//将fa,dep,son算出
    fa[u] = father;
    dep[u] = dep[father] + 1;
    sz[u] = 1;
    for (auto v : e[u]) {
        if (v == father)
            continue;
        dfs1(v, u);//顺着往下搜
        sz[u] += sz[v];
        if (sz[son[u]] < sz[v])//找出子树结点最多的那个子节点,标记为重链
            son[u] = v;
    }
}

void dfs2(int u, int t) {//算top,需要有son,所以分为2个dfs
    top[u] = t;
    if (!son[u])//u无重儿子,搜到底了,返回
        return;
    dfs2(son[u], t);
    for (int v : e[u]) {//依次搜索每一条边
        if (fa[u] == v || v == son[u])
            continue;
        dfs2(v, v);
    }
}

int lca(int u, int v) {
    while (top[u] != top[v]) {
        if(dep[top[u]] < dep[top[v]])//保证u为最深的点方便向上爬树,
            swap(u, v);            //反正找最近公共祖先,怎么换都没事
        u = fa[top[u]];
    }
    return dep[u] < dep[v] ? u : v;//谁在上面谁是爹
}

int main() {
    int n, m, s;
    cin >> n >> m >> s;
    for (int i = 0; i < n - 1; i++) {
        int x, y; cin >> x >> y;
        e[x].push_back(y);
        e[y].push_back(x);
    }
    dfs1(s, 0);
    dfs2(s, s);
    while (m--) {
        int a, b; cin >> a >> b;
        cout << lca(a, b) << '\n';
    }
    return 0;
}

  • 8
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值