Cod sursa(job #3306743)

Utilizator adimiclaus15Miclaus Adrian Stefan adimiclaus15 Data 13 august 2025 13:51:44
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.13 kb
#include <bits/stdc++.h>

using namespace std;

const int NMAX = 1e5;
int T[NMAX + 1], S[NMAX + 1][18], ad[NMAX + 1];

int query(int x, int d) {
    for(int i = log2(d); i >= 0; i--) {
        if((1 << i) <= d) {
            x = S[x][i];
            d -= (1 << i);
        }
    }
    return x;
}

int lca(int x, int y) {
    if(ad[x] < ad[y]) {
        swap(x, y);
    }
    int d = ad[x] - ad[y];
    x = query(x, d);
    if(x == y) {
        return x;
    }
    for(int i = log2(ad[x]); i >= 0; i--) {
        if(S[x][i] != S[y][i]) {
            x = S[x][i];
            y = S[y][i];
        }
    }
    return S[x][0];
}

int main() {
    ifstream cin("lca.in");
    ofstream cout("lca.out");
	int n, q;
    cin >> n >> q;
    ad[1] = 1;
    for(int i = 2; i <= n; i++) {
        cin >> T[i];
        ad[i] = ad[T[i]] + 1;
    }
    for(int i = 1; i <= n; i++) {
        S[i][0] = T[i];
    }
    for(int j = 1; j <= log2(n); j++) {
        for(int i = 1; i <= n; i++) {
            S[i][j] = S[S[i][j - 1]][j - 1];
        }
    }
    while(q--) {
        int x, y;
        cin >> x >> y;
        cout << lca(x, y) << '\n';
    }
    return 0;
}