Cod sursa(job #3307287)

Utilizator Victor5539Tanase Victor Victor5539 Data 19 august 2025 17:06:01
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.69 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>

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

const int MAX=1e5,MAXP=16;
int n,i,t[MAX+5],depth[MAX+5],node1,node2,q,v[2*MAX+5],nr,p_end[MAX+5],p_start[MAX+5],r[MAXP+5][MAX+5];
bool viz[MAX+5];
vector <int> edge[MAX+5];

bool is_ancestor(int x, int y)
{
    return p_start[x]<=p_start[y] && p_end[y]<=p_end[x];
}


void dfs(int node)
{
    v[++nr]=node;
    viz[node]=1;

    for (auto node2: edge[node])
    {
        if (!viz[node2])
        {
            depth[node2]=depth[node]+1;
            dfs(node2);
        }
    }

    v[++nr]=node;
}

int lca(int x, int y)
{
    if (is_ancestor(x,y))
        return x;

    if (is_ancestor(y,x))
        return y;

    for (int p=MAXP; p>=0; p--)
    {
        int node=r[p][x];

        if (node!=0 && !is_ancestor(node,y))
            x=node;
    }

    return r[0][x];
}

void precalc()
{
    for (i=1; i<=n; i++)
        r[0][i]=t[i];

    int p;
    for (p=1; (1<<p)<=n; p++)
    {
        for (i=1; i<=n; i++)
        {
            int node=r[p-1][i];
            r[p][i]=r[p-1][node];
        }
    }
}

int main()
{
    ios_base::sync_with_stdio(false);
    fin.tie(0); fout.tie(0);

    fin>>n>>q;

    for (i=2; i<=n; i++)
    {
        fin>>t[i];
        edge[i].push_back(t[i]);
        edge[t[i]].push_back(i);
    }

    dfs(1);

    for (i=1; i<=nr; i++)
        p_end[v[i]]=i;

    for (i=nr; i>=1; i--)
        p_start[v[i]]=i;

    precalc();

    while (q--)
    {
        fin>>node1>>node2;

        fout<<lca(node1,node2)<<'\n';
    }



    return 0;
}