Cod sursa(job #2486770)

Utilizator nicolaefilatNicolae Filat nicolaefilat Data 3 noiembrie 2019 14:25:17
Problema Lowest Common Ancestor Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.49 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <cmath>

const int MAXN = 100000 + 5;
const int MAXLOG = log2(MAXN);

using namespace std;

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

int n,intrebari;
int dp[MAXN][MAXLOG + 5],inaltime[MAXN];


int main()
{
    in>>n>>intrebari;
    inaltime[1] = 1;
    for(int i = 2; i <= n; i++){
        int nod;
        in>>nod;
        inaltime[i] = inaltime[nod] + 1;
        dp[i][0] = nod;
    }
    for(int put = 1; put < MAXLOG; put++){
        for(int i = 2; i <= n; i++){
            int stramos = dp[i][put - 1];
            if(stramos)
                dp[i][put] = dp[stramos][put - 1];
        }
    }
    for(int i = 1; i <= intrebari; i++){
        int nod1,nod2;
        in>>nod1>>nod2;
        int put = MAXLOG,stramos;
        /// eu vreau ca nod1 sa fie mai jos in graf decat nod2 adica ca inaltime[nod1] > inaltime[nod2]
        if(inaltime[nod1] < inaltime[nod2])
            swap(nod1,nod2);
        while(inaltime[nod1] != inaltime[nod2])
            nod1 = dp[nod1][0];
        while(put >= 0){
            if(dp[nod1][put] && dp[nod2][put] && dp[nod1][put] != dp[nod2][put]){
                nod1 = dp[nod1][put];
                nod2 = dp[nod2][put];
            }
            if(put == 0)
                break;
            put /= 2;
        }
        if(nod1 != nod2)
            out<<dp[nod1][0];
        else
            out<<nod1;
        out<<"\n";;
    }

    return 0;
}