Cod sursa(job #2738672)

Utilizator vlad082002Ciocoiu Vlad vlad082002 Data 6 aprilie 2021 10:55:55
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.38 kb
#include <bits/stdc++.h>
using namespace std;

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

int n, m, t[100005], lev[100005], first[100005], rmq[22][400005], logg[400005];
vector<int> g[100005], eul;

void dfs(int x, int t) {
    eul.push_back(x);
    lev[x] = lev[t]+1;
    first[x] = eul.size()-1;
    for(auto next: g[x])
        if(next != t) {
            dfs(next, x);
            eul.push_back(x);
        }
}

int lca(int x, int y) {
    int a = first[x], b = first[y];
    if(a > b) swap(a, b);

    int lg = logg[b-a+1];
    int ans = rmq[lg][a];
    if(lev[rmq[lg][b-(1<<lg)+1]] < lev[ans])
        ans = rmq[lg][b-(1<<lg)+1];

    return ans;
}

int main() {
    fin >> n >> m;
    for(int i = 2; i <= n; i++) {
        fin >> t[i];
        g[t[i]].push_back(i);
        g[i].push_back(t[i]);
    }
    dfs(1, 0);
    for(int i = 0;  i < eul.size(); i++)
        rmq[0][i] = eul[i];
    for(int i = 1; (1<<i) <= eul.size(); i++)
        for(int j = 0; j+(1<<i) <= eul.size(); j++) {
            rmq[i][j] = rmq[i-1][j];
            if(lev[rmq[i-1][j+(1<<(i-1))]] < lev[rmq[i][j]])
                rmq[i][j] = rmq[i-1][j+(1<<(i-1))];
        }

    logg[1] = 0;
    for(int i = 2; i <= eul.size(); i++)
        logg[i] = logg[i/2]+1;
    while(m--) {
        int x, y;
        fin >> x >> y;
        fout << lca(x, y) << '\n';
    }
}