Cod sursa(job #473927)

Utilizator mlazariLazari Mihai mlazari Data 1 august 2010 17:43:49
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.41 kb
#include<fstream>
#include<vector>

using namespace std;

#define NMAX 100003

ifstream fi("lca.in");

vector<int> f[NMAX],euler;
int poz[NMAX],niv[NMAX],rmq[20][NMAX<<1],log[NMAX<<1],d[20];
int n,m,x,i,a,b;

void Euler(int nod, int nv) {
  int i,s;
  poz[nod]=euler.size();
  euler.push_back(nod);
  niv[nod]=nv;
  for(i=0,s=f[nod].size();i<s;i++) {
    Euler(f[nod][i],nv+1);
    euler.push_back(nod);
  }
}

void calcD() {
  d[0]=1;
  for(int i=1;i<20;i++) d[i]=d[i-1]<<1;
}

void calcLog() {
  int i,j=1,s=euler.size();
  int dj=d[j];
  for(i=1;i<s;i++) {
    if(i==dj) {
      j++;
      dj=d[j];
    }
    log[i]=j-1;
  }
}

void preRMQ() {
  int i,j,s=euler.size(),aux1,aux2;
  for(i=0;i<s;i++) rmq[0][i]=euler[i];
  for(i=1;d[i]<=s;i++)
   for(j=0;j<=s-d[i];j++) {
     aux1=rmq[i-1][j];
     aux2=rmq[i-1][j+d[i-1]];
     if(niv[aux1]<niv[aux2])
      rmq[i][j]=aux1;
     else rmq[i][j]=aux2;
   }
}

int LCA(int a, int b) {
  a=poz[a];
  b=poz[b];
  if(a>b) swap(a,b);
  int aux1=log[b-a+1];
  int aux2=rmq[aux1][a],aux3=rmq[aux1][b-d[aux1]+1];
  if(niv[aux2]<niv[aux3]) return aux2;
  else return aux3;
}

int main() {
  fi>>n>>m;
  for(i=2;i<=n;i++) {
    fi>>x;
    f[x].push_back(i);
  }
  Euler(1,1);
  calcD();
  calcLog();
  preRMQ();
  freopen("lca.out","w",stdout);
  while(m--) {
    fi>>a>>b;
    printf("%d\n",LCA(a,b));
  }
  return 0;
}