Cod sursa(job #2576664)

Utilizator halexandru11Hritcan Alexandru halexandru11 Data 6 martie 2020 21:31:39
Problema Lowest Common Ancestor Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.11 kb
#include <bits/stdc++.h>
#define pb push_back

using namespace std;

const int nax = 1e5 + 5;

ifstream f("lca.in");
ofstream g("lca.out");

vector<int> graf[nax];
int euler[nax << 1], nivel[nax << 1], prima[nax];
int arb[nax << 1][20];
int lg[nax];
int n, m, dim;

// formez lista muchiilor de adiacenta
void Read()
{
    int i, x;
    f >> n >> m;
    for(i = 2; i <= n; ++i)
    {
        f >> x;
        graf[x].pb(i);
    }
}

// formez reprezentarea euler, nivelul fiecarui nod din reprezentarea euler
// cat si prima aparitie a fiecarui element in reprezentarea euler
void Euler(int nod, int lvl)
{
    euler[dim] = nod;
    nivel[dim] = lvl;
    prima[nod] = dim++;
    for(int& it : graf[nod])
    {
        Euler(it, lvl + 1);
        euler[dim] = nod;
        nivel[dim++] = lvl;
    }
}

// formez vectorul de logaritmi
// formez matricea arb din rmq
void PreProcessing()
{
    int i, j;
    lg[1] = 0;
    for(i = 2; i <= dim; ++i)
        lg[i] = lg[i >> 1] + 1;

    for(i = 0; i < dim; ++i)
        arb[i][0] = i;

    for(j = 1; (1 << j) <= dim; ++j)
        for(i = 0; i + (1 << j) - 1 < dim; ++i)
            if(nivel[ arb[i][j-1] ] < nivel[ arb[i + (1 << (j-1))][j-1] ])
                arb[i][j] = arb[i][j-1];
            else
                arb[i][j] = arb[i + (1 << (j-1))][j-1];
}

int Lca(int& st, int& dr)
{
    if(st > dr)
    {
        st ^= dr;
        dr ^= st;
        st ^= dr;
    }

    int range = dr - st + 1;
    int power = lg[range];
    int solutie;

    if(nivel[ arb[st][power] ] < nivel[ arb[dr - (1 << power) + 1][power] ])
        solutie = arb[st][power];
    else
        solutie = arb[dr - (1 << power) + 1][power];
    return euler[solutie];
}

int main()
{
    Read();
    Euler(1, 0);

    PreProcessing();

    int x, y;
    while(m--)
    {
        f >> x >> y;
        g << Lca(prima[x], prima[y]) << "\n";
    }

//    for(int i = 0; i < dim; ++i)
//    {
//        for(int j = 0; j <= lg[dim-i]; ++j)
//            cout << arb[i][j] << " ";
//        cout << "\n";
//    }


    return 0;
}