Cod sursa(job #2843128)

Utilizator cezar.balutaCezar Baluta cezar.baluta Data 2 februarie 2022 01:47:03
Problema Lowest Common Ancestor Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.79 kb
/*
                         __
                   _.--""  |
    .----.     _.-'   |/\| |.--.
    |    |__.-'   _________|  |_)  _______________
    |  .-""-.""""" ___,    `----'"))   __   .-""-.""""--._
    '-' ,--. `    |Cezar| .---.       |:.| ' ,--. `      _`.
     ( (    ) ) __|  7  |__\\|// _..-- \/ ( (    ) )--._".-.
      . `--' ;\__________________..--------. `--' ;--------'
       `-..-'                               `-..-'
 */
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

const int N = 1e6 + 5;
int stramosi[18][N];
int nivel[N];

void nivell(int x){
    int poz = x;
    int cnt = 0 ;
    while(x>=1) {
        if(nivel[x] != 0) {
            cnt += nivel[x];
            break;
        }
        x = stramosi[0][x];
        cnt++;
    }
    nivel[poz] = cnt;
}

int lca(int x, int y){
    int niv1 = nivel[x];
    int niv2 = nivel[y];
    int p1 = x;
    int p2 = y;
    if(niv1<niv2){
        swap(niv1,niv2);
        swap(p1,p2);
    }
    while(niv1 != niv2){
        p1 = stramosi[0][p1];
        niv1 = nivel[p1];
    }
    for(int i = 17;i>=0;i--){
        if(niv1 - (1<<i)>=1 && stramosi[i][p1]!=stramosi[i][p2]){
            p1 = stramosi[i][p1];
            p2 = stramosi[i][p2];
            niv1 = nivel[p1];
        }
    }
    if(p1==p2)
        return p1;

    p1 = stramosi[0][p1];
    return p1;

}

int main() {
    ifstream in("lca.in");
    ofstream out("lca.out");
    ios_base::sync_with_stdio(false);
    int n,m;
    in>>n>>m;
    for(int i=2;i<=n;i++){
        in>>stramosi[0][i];
    }
    for(int i=1;i<18;i++){
        for(int j=1;j<=n;j++) {
            stramosi[i][j] = stramosi[i-1][stramosi[i-1][j]];
        }
    }
    for(int i=1;i<=n;i++)
    nivell(i);
    int x,y;
    while(m--){
        in>>x>>y;
        out<<lca(x,y)<<'\n';
    }
    return 0;
}