Cod sursa(job #2096317)

Utilizator adimiclaus15Miclaus Adrian Stefan adimiclaus15 Data 28 decembrie 2017 22:43:58
Problema Lowest Common Ancestor Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.65 kb
#include <iostream>
#include <fstream>
#include <vector>
#define inf 0x3f3f3f3f
using namespace std;
ifstream f("lca.in");
ofstream g("lca.out");
int n,m,euler[400005],p[500005],niv[500005],nr,arb[800005],poz,start,finish,minim,r,t1,t2;
vector <int> a[100005];
void dfs(int nod,int nivel)
{
    euler[++nr]=nod;
    niv[nr]=nivel;
    p[nod]=nr;
    for(int i=0;i<a[nod].size();i++)
    {
        dfs(a[nod][i],nivel+1);
        euler[++nr]=nod;
        niv[nr]=nivel;
    }
}
void update(int nod ,int st,int dr)
{
    int mij;
    if(st==dr)
    {
        arb[nod]=st;
        return;
    }
    mij=(st+dr)/2;
    update(2*nod,st,mij);
    update(2*nod+1,mij+1,dr);
    if(niv[arb[2*nod]]<=niv[arb[2*nod+1]])
    {
        arb[nod]=arb[2*nod];
    }
    else
    {
        arb[nod]=arb[2*nod+1];
    }
}
void query(int nod,int st,int dr)
{
    int mij;
    if(start<=st && dr<=finish)
    {
        if(niv[arb[nod]]<minim)
        {
            minim=niv[arb[nod]];
            r=euler[arb[nod]];
        }
        return;
    }
    mij=(st+dr)/2;
    if(start<=mij)
    {
        query(2*nod,st,mij);
    }
    if(mij<finish)
    {
        query(2*nod+1,mij+1,dr);
    }
}
int main()
{
    int i,x,y;
    f>>n;
    f>>m;
    for(i=2;i<=n;i++)
    {
        f>>x;
        a[x].push_back(i);
    }
    dfs(1,0);
    update(1,1,nr);
    for(i=1;i<=m;i++)
    {
        f>>x>>y;
        t1=p[x];
        t2=p[y];
        if(t1>t2)
        {
            swap(t1,t2);
        }
        start=t1;
        finish=t2;
        minim=r=inf;
        query(1,1,nr);
        g<<r<<"\n";
    }
    return 0;
}