Cod sursa(job #992706)

Utilizator smaraldaSmaranda Dinu smaralda Data 2 septembrie 2013 14:24:33
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.36 kb
#include<stdio.h>
#define NMAX 100000
int lvl[NMAX + 5],stramos[20][NMAX + 5],tata[NMAX + 5];

int cauta (int nod, int str) {
    int p;
    for(p = 0; (1<<p) <= str; p++);
    p--;
    if((1<<p) == str)
        return stramos[p][nod];
    return cauta(stramos[p][nod],str - (1<<p));
}

int solve (int i) {
    return lvl[i] ? lvl[i] : solve(tata[i]) + 1;
}

int lca (int x, int y) {
    if(x == y)
        return x;
    if(tata[x] == tata[y])
        return tata[x];
    int p;
    for(p = 0; stramos[1<<p][x] != stramos[1<<p][y]; p++);
    --p;
    return lca(stramos[1<<p][x],stramos[1<<p][y]);
}

int main() {
    freopen("lca.in","r",stdin);
    freopen("lca.out","w",stdout);
    int j,n,x,y,i,q,m;
    scanf("%d%d",&n,&m);
    tata[1] = 1;
    for(i = 2; i <= n; i++) {
        scanf("%d",&tata[i]);
        stramos[0][i] = tata[i];
        }
    lvl[1] = 1;
    for(j = 1; (1<<j) <= n; j++)
        for(i = 1; i <= n; i++)
            stramos[j][i] = stramos[j - 1][stramos[j - 1][i]];

    for(i = 2; i <= n; i++)
        if(lvl[i] == 0)
            lvl[i] = solve(i);

    for(q = 1; q <= m; q++) {
        scanf("%d%d",&x,&y);
        if(lvl[x] < lvl[y])
            y = cauta(y,lvl[y] - lvl[x]);
        if(lvl[y] < lvl[x])
            x = cauta(x,lvl[x] - lvl[y]);
        printf("%d\n",lca(x,y));
        }
    return 0;
}