0
点赞
收藏
分享

微信扫一扫

倍增法求LCA模板

追梦人的自留地 2022-04-13 阅读 51
算法c++

倍增法求LCA

原理 树上的任意两点的最近公共祖先,是两点之间dfs序中的最小值。使用st表维护dfs序中的区间最小值。

s t [ c n t ] [ 0 ] st[cnt][0] st[cnt][0]:树的 d f s dfs dfs序,第 c n t cnt cnt个遍历的点是 s t [ c n t ] [ 0 ] st[cnt][0] st[cnt][0]

d f n [ x ] dfn[x] dfn[x]:节点编号为x的节点,第一次出现在dfs序中的位置

d e p [ x ] dep[x] dep[x]:节点x在树中的深度

模板

struct LCA {
    /*
    $st[cnt][0]$:树的$dfs$序,第$cnt$个遍历的点是$st[cnt][0]$
    $dfn[x]$:节点编号为x的节点,第一次出现在dfs序中的位置
    $dep[x]$:节点x在树中的深度
    */

    int dfn[maxn], dep[maxn], st[maxn][30], cnt;
    void dfs(int x, int fa) {
        dfn[x] = ++cnt, dep[x] = dep[fa] + 1, st[cnt][0] = x;
        for (int i = h[x]; ~i; i = ne[i]) {
            int j = e[i];
            if (j == fa) continue;
            dfs(j, x);
            st[++cnt][0] = x;
        }
    }

    void RMQ() {
        for (int j = 1; j <= 20; j++) {
            for (int i = 1; (i + (1 << j) - 1) <= cnt; i++) {
                int r = i + (1 << (j - 1));
                st[i][j] = dep[st[i][j - 1]] < dep[st[r][j - 1]] ? st[i][j - 1] : st[r][j - 1];
            }
        }
    }

    int get(int l, int r) {
        l = dfn[l], r = dfn[r];
        if (l > r) swap(l, r);
        int k = log2(r - l + 1);
        int res = dep[st[l][k]] < dep[st[r - (1 << k) + 1][k]] ? st[l][k] : st[r - (1 << k) + 1][k];
        return res;
    }

} plt;

#include <bits/stdc++.h>
#define endl "\n"
#define debug(x) cout << #x << ": -----> " << x << endl;
// typedef long long ll;
// typedef unsigned long long ull;

using namespace std;

const int maxn = 5e6 + 10;
int h[maxn], ne[maxn], e[maxn], idx;
int n, m, s;

void add(int a, int b) {
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

struct LCA {
    /*
    $st[cnt][0]$:树的$dfs$序,第$cnt$个遍历的点是$st[cnt][0]$
    $dfn[x]$:节点编号为x的节点,第一次出现在dfs序中的位置
    $dep[x]$:节点x在树中的深度
    */

    int dfn[maxn], dep[maxn], st[maxn][30], cnt;
    void dfs(int x, int fa) {
        dfn[x] = ++cnt, dep[x] = dep[fa] + 1, st[cnt][0] = x;
        for (int i = h[x]; ~i; i = ne[i]) {
            int j = e[i];
            if (j == fa) continue;
            dfs(j, x);
            st[++cnt][0] = x;
        }
    }

    void RMQ() {
        for (int j = 1; j <= 20; j++) {
            for (int i = 1; (i + (1 << j) - 1) <= cnt; i++) {
                int r = i + (1 << (j - 1));
                st[i][j] = dep[st[i][j - 1]] < dep[st[r][j - 1]] ? st[i][j - 1] : st[r][j - 1];
            }
        }
    }

    int get(int l, int r) {
        l = dfn[l], r = dfn[r];
        if (l > r) swap(l, r);
        int k = log2(r - l + 1);
        int res = dep[st[l][k]] < dep[st[r - (1 << k) + 1][k]] ? st[l][k] : st[r - (1 << k) + 1][k];
        return res;
    }

} plt;



int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> m >> s;
    memset(h, -1, sizeof h);
    for (int i = 1; i <= n - 1; i++) {
        int u, v;
        cin >> u >> v;
        add(u, v), add(v, u);
    }

    plt.dfs(s, -1);
    plt.RMQ();

    while (m--) {
        int a, b;
        cin >> a >> b;
        cout << plt.get(a, b) << endl;
    }

    return 0;
}

举报

相关推荐

0 条评论