Cod sursa(job #2976496)

Utilizator sandry24Grosu Alexandru sandry24 Data 9 februarie 2023 12:05:02
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.8 kb
#include <bits/stdc++.h>
using namespace std;
 
 
//#pragma GCC optimize("O3,unroll-loops")
//#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> pi;
#define pb push_back
#define mp make_pair
#define f first
#define s second

int MAXN = 100001;
vi euler, first(MAXN), height(MAXN);
vector<vi> adj(MAXN);

void dfs(int node, int p, int h){
    height[node] = h;
    first[node] = euler.size();
    euler.pb(node);
    for(auto i : adj[node]){
        if(i != p){
            dfs(i, node, h+1);
            euler.pb(node);
        }
    }
}

void solve(){
    int n, m;
    cin >> n >> m;
    for(int i = 0; i < n-1; i++){
        int x;
        cin >> x;
        adj[x].pb(i+2);
    }
    dfs(1, -1, 1);
    vector<vi> sparse_table(euler.size()+1, vi(25));
    for(int i = 1; i <= euler.size(); i++){
        sparse_table[i][0] = euler[i-1];
    }
    for(int j = 1; j < 25; j++){
        for(int i = 1; i + (1 << j) - 1 <= euler.size(); i++){
            int left = sparse_table[i][j-1];
            int right = sparse_table[i + (1 << (j-1))][j-1];
            sparse_table[i][j] = (height[left] < height[right]) ? left : right;
        }
    }
    for(int i = 0; i < m; i++){
        int a, b;
        cin >> a >> b;
        a = first[a]+1, b = first[b]+1;
        if(a > b)
            swap(a, b);
        int w = (int)log2(b-a+1);
        int left = sparse_table[a][w];
        int right = sparse_table[b - (1 << w) + 1][w];
        cout << ((height[left] < height[right]) ? left : right) << '\n';
    }
}
 
int main(){
    freopen("lca.in", "r", stdin);
    freopen("lca.out", "w", stdout);
    ios::sync_with_stdio(0); cin.tie(0);
    int t = 1;
    //cin >> t;
    while(t--){
        solve();
    }
}