Cod sursa(job #2866637)

Utilizator SabailaCalinSabaila Calin SabailaCalin Data 9 martie 2022 21:06:35
Problema Lowest Common Ancestor Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.79 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

const int SIZE = 200001;
const int LOG = 18;

int n, q, cnt;
int node[2*SIZE], depth[SIZE], last[SIZE];
short logarithm[SIZE];
int st[2*SIZE][LOG + 1];

vector <int> G[SIZE];

void Read()
{
    f >> n >> q;
    for (int i = 2; i <= n; i++)
    {
        int x;
        f >> x;
        G[x].push_back(i);
    }
}

void MarkAsVisited(int currentNode, int height)
{
    cnt++;
    node[cnt] = currentNode;
    depth[currentNode] = height;
    last[currentNode] = cnt;
}

void DFS(int start, int height)
{
    MarkAsVisited(start, height);
    for (unsigned int i = 0; i < G[start].size(); i++)
    {
        int neighbour = G[start][i];
        DFS(neighbour, height + 1);
        MarkAsVisited(start, height);
    }
}

void ProcessSparseTable()
{
    for (int i = 1; i <= cnt; i++)
    {
        st[i][0] = node[i];
    }

    for (int j = 1; j <= LOG; j++)
    {
        for (int i = 0; i - 1 + (1 << j) <= cnt; i++)
        {
            st[i][j] = min(st[i][j-1], st[i + (1 << (j-1))][j-1]);
        }
    }

    logarithm[1] = 0;
    for (int i = 2; i <= SIZE; i++)
    {
        logarithm[i] = logarithm[i/2] + 1;
    }
}

void ProcessQueries(int x, int y)
{
    int left = min(last[x], last[y]);
    int right = max(last[x], last[y]);
    int length = right - left + 1;

    int j = logarithm[length];
    int leftInterval = st[left][j];
    int rightInterval = st[right + 1 - (1 << j)][j];

    g << leftInterval << "\n";
}

int main()
{
    Read();
    DFS(1, 1);
    ProcessSparseTable();

    for (int i = 1; i <= q; i++)
    {
        int x, y;
        f >> x >> y;
        ProcessQueries(x, y);
    }
}