Cod sursa(job #2357705)

Utilizator crion1999Anitei cristi crion1999 Data 27 februarie 2019 17:45:19
Problema Lowest Common Ancestor Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.37 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <cmath>
#define NMAX 100005

using namespace std;
ifstream fi("lca.in");
ofstream fo("lca.out");
int N, M;
int K = 1;

vector<int> tree[NMAX];

int visited[NMAX];

vector<pair<int, int> > sparse[20];

int firstApar[NMAX];



void DFS(int node, int depth)
{
    if(visited[node])
        return;

    visited[node] = 1;
    if(firstApar[node] == 0)
        firstApar[node] = sparse[0].size();

    for(auto y : tree[node])
    {
        DFS(y, depth + 1);
        sparse[0].push_back({depth, node});
    }
}


void CreateLCASparse()
{
    sparse[0].push_back({0, 1});

    DFS(1, 0);

    K = sparse[0].size();
    for(int i = 1; i * i <= K; ++i)
        for(int j = 0; j < K - (1 << i) + 1; ++j)
            sparse[i].push_back( min(sparse[i-1][j], sparse[i-1][j + (1 << (i - 1))]));
}

int main()
{
    int a, b;
    fi >> N >> M;
    for(int i = 2; i <= N; ++i)
    {
        fi >> a;
        tree[a].push_back(i);
        tree[i].push_back(a);
    }
    CreateLCASparse();

    for(int i = 1; i <= M; ++i)
    {
        fi >> a >> b;
        a = firstApar[a];
        b = firstApar[b];
        if(a > b)
            swap(a, b);
        int lq = (int) log2(b - a + 1);
        fo << min(sparse[lq][a], sparse[lq][ b -( 1 << lq ) + 1]).second << "\n";
    }


}