Cod sursa(job #3201236)

Utilizator raresgherasaRares Gherasa raresgherasa Data 7 februarie 2024 10:45:24
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.56 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin ("lca.in");
ofstream fout ("lca.out");

const int NM = 2e5 + 5;

vector<int>g[NM];
int rmq[NM][20], sz, first[NM], k[NM], euler[NM], level[2 * NM];
int n, m;

void dfs (int nod, int dad =-1, int h = 1) {
    euler[++sz] = nod;
    level[sz] = h;
    if (first[nod] == 0) {
        first[nod] = sz;
    }
    for (int u : g[nod]) {
        if (u != dad) {
            dfs(u, nod, h + 1);
            euler[++sz] = nod;
            level[sz] = h;
        }
    }
}

void precalc() {
    dfs(1);
    for (int i = 2; i < NM; i++) {
        k[i] = 1 + k[i / 2];
    }
    for (int i = 1; i <= sz; i++) {
        rmq[i][0] = i;
    }
    for (int j = 1; j <= 20; j++) {
        for (int i = 1; i + (1 << j) - 1 <= sz; i++) {
            rmq[i][j] = rmq[i][j - 1];
            if (level[rmq[i][j]] > level[rmq[i + (1 << (j - 1))][j - 1]]) {
                rmq[i][j] = rmq[i + (1 << (j - 1))][j - 1];
            }
        }
    }
}

int solve (int l, int r) {
    l = first[l];
    r = first[r];
    if (l > r) {
        swap(l, r);
    }
    int T = k[r - l + 1];
    int answer = rmq[l][T];
    if (level[answer] > level[rmq[r - (1 << T) + 1][T]]) {
        answer = rmq[r - (1 << T) + 1][T];
    }
    return euler[answer];
}

int main() {
    fin >> n >> m;
    for (int i = 2; i <= n; i++) {
        int x; fin >> x;
        g[x].push_back(i);
    }
    precalc();
    while (m--) {
        int l, r;
        fin >> l >> r;
        fout << solve(l, r) << '\n';
    }
}