Cod sursa(job #2115892)

Utilizator radudurlesteanuDurlesteanu Radu Stefan radudurlesteanu Data 27 ianuarie 2018 11:13:48
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.12 kb
#include <fstream>
#include <vector>
#include <algorithm>
#include <math.h>
using namespace std;
ifstream fin("lca.in");
ofstream fout("lca.out");
int n,m,ct,niv;
int h[100005];
int l[200005][21];
int e[200005][21];
vector <int>V[100005];

void DFS(int x)
{ ++ct;
   e[ct][0]=x;
   if(h[x]==0)
    h[x]=ct;
    l[ct][0]=niv;

   vector<int>::iterator it;
   for(it=V[x].begin();it!=V[x].end();++it)
          {  ++niv; DFS(*it);   --niv; ++ct; e[ct][0]=x; l[ct][0]=niv;    }

}


int main()
{  fin>>n>>m;
    int i,j,x,y,sol,z;
    for(i=1;i< n;++i)
    { fin>>x;
        V[x].push_back(i+1);
    }
    DFS(1);

  for(j=1;(1<<j)<=ct;++j)
    for(i=1;i+(1<<j)-1<=ct;++i)
        if( l[i][j-1]<l[i+(1<<(j-1))][j-1] )
  { l[i][j]=l[i][j-1];
     e[i][j]=e[i][j-1];
  }
  else { l[i][j]=l[i+(1<<(j-1))][j-1];
        e[i][j]=e[i+(1<<(j-1))][j-1];
  }

  for(i=1;i<=m;++i)
  { fin>>x>>y;
      x=h[x]; y=h[y];
      if(x>y) swap(x,y);
      z=log2(y-x+1);
      if(l[x][z] <l[y-(1<<z)+1][z])
         sol=e[x][z];
      else sol=e[y-(1<<z)+1][z];
        fout<<sol<<"\n";
  }
    return 0;
}