Cod sursa(job #2953977)

Utilizator tomaionutIDorando tomaionut Data 12 decembrie 2022 20:31:25
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.36 kb
#include <bits/stdc++.h>

using namespace std;

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

int n, m, level[2000005], k;
int e[2000005], poz[100005], p2[2000005];
int rmq[22][2000005];
vector <int> a[100005];
void Dfs(int x, int niv)
{
    e[++k] = x;
    level[k] = niv;
    poz[x] = k;
    for (auto w : a[x])
    {
        Dfs(w, niv + 1);
        e[++k] = x;
        level[k] = niv;
    }
}
int main()
{
    int i, j, x, lg;
    fin >> n >> m;
    for (i = 2; i <= n; i++)
    {
        fin >> x;
        a[x].push_back(i);
    }

    Dfs(1, 0);

    p2[1] = 0;
    for (i = 2; i <= k; i++)
        p2[i] = p2[i / 2] + 1;

    for (i = 1; i <= k; i++)
        rmq[0][i] = i;

    for (i = 1; i <= p2[k]; i++)
        for (j = 1; j <= k - (1 << i) + 1; j++)
        {
            lg = (1 << (i - 1));
            rmq[i][j] = rmq[i - 1][j];
            if (level[rmq[i - 1][j]] > level[rmq[i - 1][j + lg]])
                rmq[i][j] = rmq[i - 1][j + lg];
        }

    while (m--)
    {
        fin >> i >> j;
        i = poz[i];
        j = poz[j];
        if (i > j)
            swap(i, j);
        lg = p2[j - i + 1];
        if (level[rmq[lg][i]] < level[rmq[lg][j - (1 << lg) + 1]])
            fout << e[rmq[lg][i]] << "\n";
        else fout << e[rmq[lg][j - (1 << lg) + 1]] << "\n";
    }

    return 0;
}