Cod sursa(job #3326149)

Utilizator TeodorVTeodorV TeodorV Data 27 noiembrie 2025 15:26:43
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.54 kb
#include <bits/stdc++.h>

using namespace std;

const string NUMEFISIER="lca";
ifstream fin(NUMEFISIER+".in");
ofstream fout(NUMEFISIER+".out");

const int Nmax=100001;
const int Lmax=17;
int stramosi[Nmax][Lmax];
vector<int> fii[Nmax];
int n;
void initStramosi(){
    for(int p2=1; (1<<p2)<=n; p2++)
    {
        for(int j=1; j<=n; j++)
        {
            /*if(stramosi[j][p2-1]==0)
                stramosi[j][p2]=0;
            else */
                stramosi[j][p2]=stramosi[stramosi[j][p2-1]][p2-1];
        }
    }
}
pair<int, int> timpDfs[Nmax];
void dfs(int node){
    static int t=0;

    timpDfs[node].first=++t;
    for(int fiu : fii[node])
    {
        dfs(fiu);
    }
    timpDfs[node].second=++t;
}
bool esteStramos(int stramos, int node){
    return (timpDfs[stramos].first<=timpDfs[node].first && timpDfs[node].second<=timpDfs[stramos].second);
}

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

    int pas=Lmax-1;
    while(pas>=0)
    {
        if(stramosi[x][pas]!=0 && !esteStramos(stramosi[x][pas], y))
            x=stramosi[x][pas];
        pas--;
    }
    return stramosi[x][0];//x cel mai indepartat al lui x initial care NU este stramos al lui y
}

int main()
{
    int m;
    fin>>n>>m;
    for(int i=2; i<=n; i++)
    {
        fin>>stramosi[i][0];
        fii[stramosi[i][0]].push_back(i);
    }
    initStramosi();
    dfs(1);
    for(int i=1; i<=m; i++)
    {
        int x,y;
        fin>>x>>y;
        fout<<lca(x, y)<<'\n';
    }
    return 0;
}