Cod sursa(job #1140778)

Utilizator alecsandrualex cuturela alecsandru Data 12 martie 2014 11:40:11
Problema Lowest Common Ancestor Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.3 kb
#include<cstdio>
#include<vector>
#include<cmath>
using namespace std;
int parcurgere[20][5000000],lim,j,nr,n,m,poz[200100];
vector <int> vecini[200100];
void dfs(int tata)
{
    int i,fiu;
    parcurgere[1][++nr]=tata;
    poz[tata]=nr;
    for(i=0;i<vecini[tata].size();i++)
    {
        fiu=vecini[tata][i];
        dfs(fiu);
        parcurgere[1][++nr]=tata;
    }
}
int main()
{
    int x,i,k;
    freopen("lca.in","r",stdin);
    freopen("lca.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(i=1;i<n;i++)
    {
        scanf("%d",&x);
        vecini[x].push_back(i+1);
    }
    dfs(1);
    parcurgere[1][0]=nr;
    i=2;
    for(k=2;i<=nr;k++)
    {
        lim=parcurgere[k-1][0]-i/2;
        for(j=1;j<=lim;j++)
        {
            parcurgere[k][j]=min(parcurgere[k-1][j],parcurgere[k-1][j+i/2]);
           // printf("%d ",parcurgere[k][j]);
        }
        nr=lim;
        parcurgere[k][0]=lim;
       // printf("\n");
        i=1<<k;
    }
    int a,b,y,rez,p;
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&a,&b);
        x=min(poz[a],poz[b]);
        y=max(poz[a],poz[b]);
        lim=y-x-1;
        lim=log2(lim);
        p=1<<lim;
        rez=min(parcurgere[lim+1][x],parcurgere[lim+1][y-p]);
        printf("%d\n",rez);
    }
    return 0;
}