Cod sursa(job #777984)

Utilizator ionut_blesneagIonut Blesneag ionut_blesneag Data 13 august 2012 19:16:46
Problema Lowest Common Ancestor Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.42 kb
/*Lowest common ancesor*/
#include<cstdio>
#include<list>
#include<algorithm>
using namespace std;

list<int> L[100001];
int n,m,t,level[100001],size_eul,i,j,k,dif,rez;
int RMQ[20][200001], log[100001];

int minimum(int a, int b)
{if(level[a]<level[b])
  return a;
return b;}

int maximum(int a, int b)
{if(level[a]>level[b])
  return a;
return b;}

void parcurgere(int x, int lvl)
{list<int>::iterator it;
for(it=L[x].begin(); it!=L[x].end(); it++)
{size_eul++;
 RMQ[0][size_eul]=x;    //eulerian representation
 parcurgere(*it,lvl+1);}

size_eul++;
RMQ[0][size_eul]=x;  
}

int main()
{freopen("lca.in","r",stdin);
freopen("lca.out","w",stdout);
scanf("%d %d",&n,&m);
for(i=2; i<=n; i++)
  {scanf("%d",&t);
   level[i]=level[t]+1;
   L[t].push_back(i);}
parcurgere(1,1);   

for(i=2; i<=size_eul; i++)
  log[i]=log[i/2]+1;
  

i=1;  
while((1<<i)<=size_eul)
{
for(j=1; j<=(size_eul-(1<<i)+1); j++)
 RMQ[i][j]=minimum(RMQ[i-1][j], RMQ[i-1][j+(1<<(i-1))]);
i++; 
}  

int x,y,gx,gy,pozx,pozy;

for(i=1; i<=m; i++)
{scanf("%d %d",&x,&y);
gx=0;  gy=0;
for(j=1; j<=size_eul; j++)
  {if(gx==0 && RMQ[0][j]==x)
     {pozx=j;   gx=1;}
  if(gy==0 && RMQ[0][j]==y)
     {pozy=j;   gy=1;}   
  if(gx==1 && gy==1)
    break;}   

dif=max(pozx,pozy)-min(pozx,pozy);
k=log[dif+1];
rez=minimum(RMQ[k][min(pozx,pozy)],RMQ[k][max(pozx,pozy)-(1<<k)+1]);
printf("%d\n",rez);  
       
}
   
return 0;}