Cod sursa(job #996474)

Utilizator assa98Andrei Stanciu assa98 Data 11 septembrie 2013 23:57:42
Problema Lowest Common Ancestor Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.21 kb
#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;

const int MAX_N=100100;
const int MAX_LG=17;

int d[MAX_LG][MAX_N];
int lv[MAX_N];

vector<int> A[MAX_N];

int query(int x,int y) {
    if(lv[x]<lv[y])
        swap(x,y);

    for(int i=MAX_LG-1;i>=0;i--) {
        if((lv[x]-(1<<i))>=lv[y]) {
            x=d[i][x];
        }
    }
    //fprintf(stderr,"%d %d\n",lv[x],lv[y]);
    if(x==y)
        return x;
    for(int i=MAX_LG-1;i>=0;i--) {
        if(d[i][x] && d[i][x] != d[i][y]) {
            x=d[i][x];
            y=d[i][y];
        }
    }

    return d[0][x];
}

void dfs(int nod,int lev) {
    lv[nod]=lev;
    for(vector<int>::iterator it=A[nod].begin();it!=A[nod].end();it++)
        dfs(*it,lev+1);
}

int main() {
    freopen("lca.in","r",stdin);
    freopen("lca.out","w",stdout);
    
    int n,m;
    scanf("%d%d",&n,&m);

    for(int i=2;i<=n;i++) {
        scanf("%d",&d[0][i]);
        A[d[0][i]].push_back(i);
    }

    for(int k=1;(1<<k)<n;k++) {
        for(int i=1;i<=n;i++) {
            d[k][i]=d[k-1][d[k-1][i]];
        }
    }
    
    dfs(1,1);

    for(int i=1;i<=m;i++) {
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d\n",query(x,y));
    }
    return 0;
}