Cod sursa(job #2973238)

Utilizator divadddDavid Curca divaddd Data 31 ianuarie 2023 16:17:45
Problema Lowest Common Ancestor Scor 30
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.62 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <cstring>
#include <cmath>
#define int long long
#define NMAX  100002
#define LGMAX 20
using namespace std;
int n,x,y,m,euler[NMAX*2],k,niv[NMAX],rmq[LGMAX][NMAX],mbit[NMAX*2];
vector<int> v[NMAX];
pair<int, int> ap[NMAX];

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

void dfs(int nod, int tata){
    euler[++k] = nod;
    ap[nod] = {k, k};
    for(int vecin: v[nod]){
        if(niv[vecin] == 0){
            niv[vecin] = 1+niv[nod];
            dfs(vecin, nod);
            euler[++k] = nod;
            ap[nod].second = k;
        }
    }
}

int lca(int a, int b){
    int st = ap[a].first;
    int dr = ap[b].second;
    if(st > dr){
        swap(a, b);
        st = ap[a].first;
        dr = ap[b].second;
    }
    int len = dr-st+1;
    int log = log2(len);
    int x = rmq[log][st], y = rmq[log][dr-(1<<log)+1];
    if(niv[x] < niv[y]){
        return x;
    }
    return y;
}

signed main()
{
    fin >> n >> m;
    for(int i = 2; i <= n; i++){
        fin >> x;
        v[x].push_back(i);
        v[i].push_back(x);
    }
    niv[1] = 1;
    dfs(1, -1);
    for(int i = 1; i <= k; i++){
        rmq[0][i] = euler[i];
    }
    int lg = log2(k);
    for(int i = 1; i <= lg; i++){
        for(int j = 1; j+(1<<(i-1)) <= k; j++){
            rmq[i][j] = rmq[i-1][j];
            if(niv[rmq[i-1][j+(1<<(i-1))]] < niv[rmq[i][j]]){
                rmq[i][j] = rmq[i-1][j+(1<<(i-1))];
            }
        }
    }
    while(m--){
        fin >> x >> y;
        fout << lca(x, y) << "\n";
    }
    return 0;
}