Cod sursa(job #3042120)

Utilizator Chris_BlackBlaga Cristian Chris_Black Data 3 aprilie 2023 22:46:18
Problema Lowest Common Ancestor Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include <fstream>
#include <vector>


using namespace std;
ifstream cin("lca.in");
ofstream cout("lca.out");
const int N = 1e5 + 9, LG = 18;
int n,m,x,a,b;
int t[N][LG],dep[N];
vector<vector<int>> G;

void Dfs(int x,int p)
{
    t[x][0] = p;
    for(int i = 1; i < LG; ++i)
        t[x][i] = t[t[x][i-1]][i-1];

    for(auto y : G[x])
    {
        dep[y] = dep[x] + 1;
        Dfs(y,x);
    }
}

int get_lca(int a,int b)
{
    if(dep[a] < dep[b])
        swap(a,b);

    for(int i = LG - 1; i >= 0; --i)
        if(dep[t[a][i]] >= dep[b])
            a = t[a][i];

    for(int i = LG - 1; i >= 0; --i)
        if(t[a][i] != t[b][i])
        {
            a = t[a][i];
            b = t[b][i];
        }

    if(a != b)
        return t[a][0];
    return a;
}

int main()
{
    cin>>n>>m;
    G = vector<vector<int>>(n+1);
    for(int i = 2; i <= n; ++i)
    {
        cin>>x;
        G[x].push_back(i);
    }

    Dfs(1,0);

    while(m--)
    {
        cin>>a>>b;
        cout<<get_lca(a,b)<<'\n';
    }

    return 0;
}