Cod sursa(job #1309864)

Utilizator thesvcoolmanLucian Bicsi thesvcoolman Data 6 ianuarie 2015 09:32:33
Problema Lowest Common Ancestor Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.06 kb
#include<cstdio>
#include<vector>

#define NMAX 100002

using namespace std;

vector<int> PARENT;
vector<int> RANK;
vector<int> PARENT2;
int n, t;


int lca (int a, int b) {
    if(RANK[a] > RANK[b]) return lca (b, a);

    while(RANK[a] < RANK[b] / 2) {
        b = PARENT2[b];
    }
    while(RANK[a] < RANK[b]) {
        b = PARENT[b];
    }

    while(b != a) {
        a = PARENT[a];
        b = PARENT[b];
    }

    return a;
}

int nth_parent(int node, int n) {
    while(RANK[node] > n) node = PARENT[node];
    return node;
}

int main() {
    freopen("lca.in", "r", stdin);
    freopen("lca.out", "w", stdout);

    scanf("%d %d", &n, &t);

    PARENT.resize(n+1, 0);
    RANK.resize(n+1, 0);
    PARENT2.resize(n+1, 0);

    for(int i=2; i<=n; i++) {
        scanf("%d", &PARENT[i]);
        RANK[i] = RANK[PARENT[i]] + 1;
        PARENT2[i] = nth_parent(i, RANK[i]/2);
    }

    int a, b;
    for(int i=1; i<=t; ++i) {
        scanf("%d %d", &a, &b);
        printf("%d\n", lca(a, b));
    }

    return 0;
}