Cod sursa(job #2286283)

Utilizator marian013Giugioiu Marian Constantin marian013 Data 20 noiembrie 2018 00:23:04
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.02 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream f("lca.in");
ofstream g("lca.out");
int n,m,k,prim[100005],rmq[19][100005<<1],lg[100005<<1];
vector <int> a[1<<17];
void citire(){
    f>>n>>m;
    int x,i;
    for(i=2;i<=n;i++)
    {
        f>>x;
        a[x].push_back(i);
    }
}
void rec(int nod){
    rmq[0][++k]=nod;
    prim[nod]=k;
    for(int i=0;i<(int)a[nod].size();i++)
    {
        rec(a[nod][i]);
        rmq[0][++k]=nod;
    }
}
void calcrmq()
{
    int i,j;
    for(i=2;i<=k;i++)
        lg[i]=lg[i/2]+1;
    for(i=1;(1<<i)<=k;i++)
        for(j=1;j<=k-(1<<(i-1));j++)
        rmq[i][j]= min( rmq[i-1][j] , rmq[i-1][j+(1<<(i-1))] );
}
int lca(int x, int y){
	int a1 = prim[x], b1 = prim[y];
	if(a1 > b1) swap(a1, b1);
	int l = lg[b1 - a1 + 1];
	return min(rmq[l][a1],rmq[l][b1-(1<<l)+1]);
}
int main()
{
    int x,y,i;
    citire();
    rec(1);
    calcrmq();
    for(i=1;i<=m;i++)
    {
        f>>x>>y;
        g<<lca(x,y)<<"\n";
    }
    return 0;
}