RMQ求LCA

題目連接

rmq求LCA，interesting。

一直沒有學這玩意兒是由於CTSC的Day1T2，當時我打的樹剖LCA 65分，gxb打的rmq LCA 45分。。。

不過rmq理論複雜度仍是小一點的，就學一下把。

RMQ求LCA

咱們要用到三個數組

\(dfn[i]\):第\(i\)個節點位置的時間戳

\(id[i][j]\)：在歐拉序中\(i\)到\(i + 2^j - 1\)這段區間內深度最小的節點編號

\(dep[i]\)：第\(i\)個節點的深度

實際上用到了一個性質：

對於任意兩點的\(LCA\)，必定是它們歐拉序中兩點之間的最小值

歐拉序是什麼？就是把dfs中遍歷到每個一個節點(包括回溯時遍歷到)加到一個序列裏，最終獲得的就是歐拉序

時空複雜度

設\(T = 2 * n - 1\)

時間複雜度：

預處理：\(O(TlogT)\)

查詢：\(O(1)\)

空間複雜度：

考慮歐拉序中有多少個點，首先每一個點被訪問到的時候會作出\(1\)的貢獻

其次在遍歷每條邊時會多出\(1\)的共貢獻

所以總空間複雜度爲：\(O(T)\)

// luogu-judger-enable-o2
#include<bits/stdc++.h>
const int MAXN = 1e6 + 10;
using namespace std;
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}
int N, Q, S, tot, dfn[MAXN], rev[MAXN], dep[MAXN], id[MAXN][21], lg2[MAXN], rd[MAXN];
vector<int> v[MAXN];
void dfs(int x, int fa) {
    dfn[x] = ++tot; dep[x] = dep[fa] + 1; id[tot][0] = x; 
    for(int i = 0, to; i < v[x].size(); i++) {
        if((to = v[x][i]) == fa) continue;
        dfs(to, x);
        id[++tot][0] = x;
    }
}
void RMQ() {
    for(int i = 2; i <= tot; i++) lg2[i] = lg2[i >> 1] + 1;
    for(int j = 1; j <= 20; j++) {
        for(int i = 1; (i + (1 << j) - 1) <= tot; i++) {
            int r = i + (1 << (j - 1));
            id[i][j] = dep[id[i][j - 1]] < dep[id[r][j - 1]] ? id[i][j - 1] : id[r][j - 1];
        }
    }
}
int Query(int l, int r) {
    if(l > r) swap(l, r);
    int k = lg2[r - l + 1];
    return dep[id[l][k]] < dep[id[r - (1 << k) + 1][k]] ? id[l][k] : id[r - (1 << k) + 1][k];
}
int main() {
    freopen("a.in", "r", stdin);
    N = read(); Q = read(); S = read();
    for(int i = 1; i <= N - 1; i++) {
        int x = read(), y = read();
        v[x].push_back(y); v[y].push_back(x);
    }
    dfs(S, 0);
    RMQ();
    while(Q--) {
        int x = read(), y = read();
        printf("%d\n", Query(dfn[x], dfn[y]));
    }
    return 0;
}