Cod sursa(job #1021136)

Utilizator cat_red20Vasile Ioana cat_red20 Data 3 noiembrie 2013 12:29:02
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.78 kb
#include<fstream>
#include<vector>
using namespace std;
int p2[20],n,m,niv[20][200000],eul[20][200000],t[100001],u,poz[100001];
vector <int>fii[100001];
ifstream fin("lca.in");
ofstream fout("lca.out");

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

void citire()
{
    fin>>n>>m;
    for(int i=2;i<=n;i++)
    {
        fin>>t[i];
        fii[t[i]].push_back(i);
    }
}

void dfs(int nod,int lvl)
{
    eul[0][++u]=nod;
    niv[0][u]=lvl;
    poz[nod]=u;
    for(int i=0;i<fii[nod].size();i++)
    {
        dfs(fii[nod][i],lvl+1);
        eul[0][++u]=nod;
        niv[0][u]=lvl;
    }
}

void rmq()
{
    for(int i=1;p2[i]<=u;i++)
    {
        for(int j=p2[i];j<=u;j++)
        {
            if(niv[i-1][j]<niv[i-1][j-p2[i-1]])
            {
                niv[i][j]=niv[i-1][j];
                eul[i][j]=eul[i-1][j];
            }
            else
            {
                niv[i][j]=niv[i-1][j-p2[i-1]];
                eul[i][j]=eul[i-1][j-p2[i-1]];
            }
        }
    }
}

void queries()
{
    int x,y,l,j,st,dr;
    for(int i=1;i<=m;i++)
    {
        fin>>x>>y;
        if(poz[x]>poz[y])
        {
            l=poz[x]-poz[y]+1;
            st=poz[y];
            dr=poz[x];
        }
        else
        {
            l=poz[y]-poz[x]+1;
            st=poz[x];
            dr=poz[y];
        }
        j=0;
        while(l>=p2[j])
        {
            j++;
        }
        j--;
        if(niv[j][dr]>niv[j][st+p2[j]-1])
        {
            fout<<eul[j][st+p2[j]-1]<<"\n";
        }
        else
            fout<<eul[j][dr]<<"\n";
    }
}

int main()
{
    init();
    citire();
    dfs(1,1);
    rmq();
    queries();
    return 0;
}