Cod sursa(job #1415596)

Utilizator witselWitsel Andrei witsel Data 5 aprilie 2015 15:09:01
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.45 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <cmath>
using namespace std;
 
ifstream fin("lca.in");
ofstream fout("lca.out");
 
const int nmax=100001;
int Euler[2*nmax],Level[2*nmax],H[nmax],n,m,k=1,M[2*nmax][19];
vector<int> Graf[nmax];
 
void ST(int L[])
{
    k--;
    for(int i=1;i<=k;++i)
        M[i][0]=i;
 
    for(int j=1; (1<<j) <k;++j)
        for(int i=1;i+(1<<j)<=k;++i)
    {
        if(L[M[i][j-1]]<=L[M[i+(1<<(j-1))][j-1]])
            M[i][j]=M[i][j-1];
        else
            M[i][j]=M[i+(1<<(j-1))][j-1];
    }
}
void parcurgere(int nodCurent,int level)
{
    Level[k]=level;
    Euler[k]=nodCurent;
    H[nodCurent]=k;
    k++;
    for(int i=0;i<Graf[nodCurent].size();++i)
    {
        parcurgere(Graf[nodCurent][i],level+1);
        Level[k]=level;
        Euler[k]=nodCurent;
        k++;
    }
 
}
 
void citire()
{
    int x;
    fin>>n>>m;
    for(int i=2;i<=n;++i)
    {
        fin>>x;
        Graf[x].push_back(i);
    }
}
 
 
int RMQ(int x,int y)
{
    if(x>y)
        swap(x,y);
    int k=log2(y-x+1);
    if(Level[M[x][k]]<=Level[M[y-(1<<k)+1][k]])
        return M[x][k];
    else
        return M[y-(1<<k)+1][k];
}
void query()
{
    int x,y;
    for(int i=1;i<=m;++i)
    {
        fin>>x>>y;
        fout<<Euler[RMQ(H[x],H[y])]<<" \n";
    }
}
int main()
{
    citire();
    parcurgere(1,0);
    ST(Level);
    query();
 
    return 0;
}