Cod sursa(job #2565420)

Utilizator radugnnGone Radu Mihnea radugnn Data 2 martie 2020 14:12:05
Problema Lowest Common Ancestor Scor 30
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.55 kb
#include <bits/stdc++.h>
#define DIM 100010
using namespace std;
ifstream fin("lca.in");
ofstream fout("lca.out");
vector<int> L[DIM];
int E[2*DIM],p[30],lg[DIM+10],t[DIM],aparitie[DIM];
int k,i,n,m,a,b;
pair<int,int> D[20][2*DIM];
void dfs(int nod, int niv){
    D[0][++k].first=niv;
    D[0][k].second=nod;
    aparitie[nod]=k;
    for(int i=0;i<L[nod].size();i++){
        int vecin=L[nod][i];
        dfs(vecin,niv+1);
        D[0][++k].first=niv;
        D[0][k].second=nod;

    }
}
void puteri(){
    p[0]=1;
    for(int i=1;p[i-1]<=k;i++)
        p[i]=2*p[i-1];
}
void logaritm(){
    for(int i=2;i<=k;i++){
        lg[i]=1+lg[i/2];
    }
}
void dinamica(){
    for(int i=1;p[i]<=k;i++){
       for(int j=1;j<=k;j++){
            if(j+p[i-1]<=k){
                if(D[i-1][j].first < D[i-1][j+p[i-1]].first)
                    D[i][j]=D[i-1][j];
                else
                    D[i][j]=D[i-1][j+p[i-1]];
            }
            else
                D[i][j]=D[i-1][j];
       }
    }
}
int rmq(int a,int b){
    a=aparitie[a];
    b=aparitie[b];
    if(a>b)
        swap(a,b);
    if(D[lg[b-a]][a].first < D[lg[b-a]][b-p[lg[b-a]]+1].first)
        return D[lg[b-a]][a].second;
    else
        return D[lg[b-a]][b-p[lg[b-a]]+1].second;
}
int main(){
    fin>>n>>m;
    for(i=2;i<=n;i++){
        fin>>t[i];
        L[t[i]].push_back(i);
    }
    dfs(1,0);
    puteri();
    logaritm();
    dinamica();
    for(i=1;i<=m;i++){
        fin>>a>>b;
        fout<<rmq(a,b)<<"\n";
    }
    return 0;
}