Cod sursa(job #2283277)

Utilizator marian013Giugioiu Marian Constantin marian013 Data 15 noiembrie 2018 12:17:35
Problema Lowest Common Ancestor Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.39 kb
#include <fstream>
using namespace std;
ifstream f("lca.in");
ofstream g("lca.out");
#define min(a,b) (a<b?a:b)
int n,m,c[1000],a[100005][1000],k,b[100005<<1],lev[100005<<1],prim[100005],rmq[20][100005<<1],lg[100005<<1];
void citire(){
    f>>n>>m;
    int x,i;
    for(i=2;i<=n;i++)
    {
        f>>x;
        a[x][++c[x]]=i;
    }

}
void rec(int nod, int lv){
    b[++k]=nod;
    lev[k]=lv;
    prim[nod]=k;
    for(int i=1;i<=c[nod];i++)
    {
        rec(a[nod][i],lv+1);
        b[++k]=nod;
        lev[k]=lv;
    }
}
void calcrmq()
{
    int i,j,l;
    lg[1]=0;
    for(i=2;i<=k;i++)
        lg[i]=lg[i/2]+1;
    for(i=1;i<=k;i++)
        rmq[0][i]=b[i];
    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 dif1,l,sol,dif2 ;
	int a1 = prim[x], b1 = prim[y];
	if(a1 > b1) swap(a1, b1);
	dif1 = b1 - a1 + 1;
	l = lg[dif1];
	sol = rmq[l][a1];
	dif2 = dif1 - (1 << l);
	if(lev[sol] > lev[rmq[l][a1 + dif2]])
		sol = rmq[l][a1 + dif2];
	return sol;
}
int main()
{
    int x,y,i,j;
    citire();
    rec(1,1);
    calcrmq();
/*    for(i=0;(1<<i)<=k;i++)
    {
        for(j=1;j<=k-(1<<i)+1;j++)
            g<<rmq[i][j]<<" ";
        g<<"\n";
    }*/
    for(i=1;i<=m;i++)
    {
        f>>x>>y;
        g<<lca(x,y)<<"\n";
    }
    return 0;
}