Cod sursa(job #608146)

Utilizator Magnuscont cu nume gresit sau fals Magnus Data 15 august 2011 11:48:39
Problema Lowest Common Ancestor Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.02 kb
#include <fstream>

using namespace std;

int n,m,a[100001],x,y,i,j,p[100001][20],l[100001];

int lca(int n,int x,int y)
{
    int i,aux,lg;
    if(l[x]<l[y])
    {
        aux=x;
        x=y;
        y=aux;
    }
    for(lg=1;(1<<lg)<=l[x];++lg);
    --lg;
    for(i=lg;i>=0;--i)
        if(l[x]-(1<<i)>=l[y])
            x=p[x][i];
    if(x==y)
        return x;
    for(i=lg;i>=0;--i)
        if(p[x][i]!=-1&&p[x][i]!=p[y][i]&&p[y][i]!=-1)
            x=p[x][i],y=p[y][i];
    return a[x];
}

int main()
{
    ifstream fin("lca.in");
    ofstream fout("lca.out");
    fin>>n>>m;
    l[1]=1;
    for(i=2;i<=n;++i)
    {
        fin>>a[i];
        l[i]=l[a[i]]+1;
    }
    for(i=0;i<=n;++i)
    for(j=1;j<=18;++j)
        p[i][j]=-1;
    for(i=1;i<=n;++i)
        p[i][0]=a[i];
    for(j=1;j<=18;++j)
    for(i=0;i<=n;++i)
        if(p[i][j-1]!=-1)
            p[i][j]=p[p[i][j-1]][j-1];
    for(;m;--m)
    {
        fin>>x>>y;
        fout<<lca(n,x,y)<<"\n";
    }
    return 0;
}