Cod sursa(job #1804449)

Utilizator iulianrotaruRotaru Gheorghe-Iulian iulianrotaru Data 12 noiembrie 2016 16:31:30
Problema Lowest Common Ancestor Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.86 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("lca.in");
ofstream g("lca.out");
int n,m,i,j,x,y,nr,d,k,p[1<<17],put[1<<18],rmq[19][1<<18];
vector <int> G[1<17];
void dfs(int x)
{
    rmq[0][++nr]=x;
    if(!p[x]) p[x]=nr;
    for(int i=0;i<G[x].size();++i)
    {
        dfs(G[x][i]);
        rmq[0][++nr]=x;
    }
}
int main()
{
    f>>n>>m;
    for(i=2;i<=n;++i)
    {
        f>>x;
        G[x].push_back(i);
    }
    dfs(1);
    /// prima putere de 2 mai mica ca i
    put[1]=0;
    for(i=2;i<=nr;i++) put[i]=put[i/2]+1;
    for(i=1;(1<<i)<=nr;++i)
        for(j=1,k=1<<(i-1);j<=nr;++j)
             rmq[i][j]=min(rmq[i-1][j],rmq[i-1][j+k]);
    for(i=1;i<=m;i++)
    {
        f>>x>>y;
        if(p[x]>p[y]) swap(x,y);
        d=put[p[y]-p[x]];
        g<<min(rmq[d][p[x]],rmq[d][p[y]-(1<<d)+1])<<'\n';
    }
    return 0;
}