Cod sursa(job #1401754)

Utilizator 0051David Sera 0051 Data 26 martie 2015 09:24:09
Problema Lowest Common Ancestor Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.49 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <cmath>

using namespace std;

#define MAX 100002

ifstream fin("lca.in");
ofstream fout("lca.out");

typedef vector < int > ::iterator iter;
vector < int > g[MAX];
int b[2*MAX-1],lev[2*MAX-1],r[17][MAX],n,m,k;

void df(int nod, int niv)
{
    for(iter it=g[nod].begin();it!=g[nod].end();it++)
    {
        b[++k]=nod;
        lev[k]=niv;
        df(*it,niv+1);
    }
    b[++k]=nod;
    lev[k]=niv;
}

void rmq()
{
    for(int i=1;i<=k;i++)
        r[0][i]=lev[i];
    for(int i=1;(1<<i)<=k;i++)
        for(int j=1;j<=k-(1<<i)+1;j++)
            r[i][j]=min(r[i-1][j], r[i-1][j+(1<<(i-1))]);
}

int main()
{
    fin>>n>>m;
    for(int i=2; i<=n; i++)
    {
        int x;
        fin>>x;
        g[x].push_back(i);
    }
    df(1,0);
    rmq();
    while(m--)
    {
        int x,y,start,stop;
        fin>>x>>y;
        for(int i=1;i<=k;i++)
            if(b[i]==x)
            {
                start=i;
                break;
            }
        for(int i=start;i<=k;i++)
            if(b[i]==y)
            {
                stop=i;
                break;
            }
        int h = log2(stop-start+1);
        int l=min(r[h][start], r[h][start+(stop-start+1)-(1<<h)]);
        for(int i=start;i<=stop;i++)
            if(lev[i]==l)
            {
                fout<<b[i]<<"\n";
                break;
            }
    }
    fin.close();
    fout.close();
    return 0;
}