Cod sursa(job #2959402)

Utilizator domistnSatnoianu Dominic Ioan domistn Data 30 decembrie 2022 23:27:27
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.86 kb
#include <bits/stdc++.h>

using namespace std;

const int NMAX = 1e5, LGMAX = 18;

int n, q, etl, lg2[2 * NMAX + 1],
    first[NMAX + 1], niv[NMAX + 1], sparse[2 * NMAX + 1][LGMAX + 1];
vector<int> adj[NMAX + 1], eulerTour;

void precalcLg2() {
    lg2[1] = 0;
    for(int i = 2; i <= 2 * NMAX; ++i)
        lg2[i] = lg2[i / 2] + 1;
}

void constructEulerTour(int nod = 1, int par = -1, int nivCrt = 0) {
    niv[nod] = nivCrt;
    first[nod] = eulerTour.size();
    eulerTour.push_back(nod);
    for(const auto &el : adj[nod]) {
        if(el == par) continue;
        constructEulerTour(el, nod, nivCrt + 1);
        eulerTour.push_back(nod);
    }
}

void constructSparse() {
    for(int i = 1; i <= etl; ++i)
        sparse[i][0] = eulerTour[i];
    for(int sz = 1; (1 << sz) <= etl; ++sz)
        for(int i = 1; i + (1 << sz) - 1 <= etl; ++i) {
            const int v1 = sparse[i][sz - 1],
                      v2 = sparse[i + (1 << (sz - 1))][sz - 1];
            if(niv[v1] < niv[v2]) sparse[i][sz] = v1;
            else sparse[i][sz] = v2;
        }
}

int lca(int x, int y) {
    x = first[x];
    y = first[y];
    if(x > y) swap(x, y);
    const int lg2dist = lg2[y - x + 1];
    const int v1 = sparse[x][lg2dist],
              v2 = sparse[y - (1 << lg2dist) + 1][lg2dist];
    if(niv[v1] < niv[v2]) return v1;
    return v2;
}

int main()
{
    precalcLg2();
    freopen("lca.in", "r", stdin);
    freopen("lca.out", "w", stdout);
    scanf("%d%d", &n, &q);
    for(int i = 2, x; i <= n; ++i) {
        scanf("%d", &x);
        adj[i].push_back(x);
        adj[x].push_back(i);
    }
    eulerTour.push_back(0);
    constructEulerTour();
    etl = eulerTour.size() - 1;
    constructSparse();
    for(int i = 1, x, y; i <= q; ++i) {
        scanf("%d%d", &x, &y);
        printf("%d\n", lca(x, y));
    }
    return 0;
}