Cod sursa(job #3137064)

Utilizator DariusGhercaDarius Gherca DariusGherca Data 10 iunie 2023 21:06:20
Problema Lowest Common Ancestor Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.71 kb
#include <fstream>
#include <iostream>
using namespace std;
ifstream f("lca.in");
ofstream g("lca.out");
const int N = 1e5 + 10;
int v[N], h[N];
int solve(int a, int b){
    while(h[a] > h[b]){
        a = v[a];
    }
    while(h[b] > h[a]){
        b = v[b];
    }
    while(v[a] != v[b]){
        a = v[a];
        b = v[b];
    }
    if(a == b)
        return a;
    else
        return v[a];
}
int main()
{
    int n, m;
    f >> n >> m;
    h[1] = 0;
    for(int i = 2; i <= n; i++){
        int x;
        f >> x;
        v[i] = x;
        h[i] = h[x] + 1;
    }
    int a, b;
    while(m){
        f >> a >> b;
        g << solve(a, b) << "\n";
        m--;
    }
    return 0;
}