Cod sursa(job #1790983)

Utilizator stelian2000Stelian Chichirim stelian2000 Data 28 octombrie 2016 22:24:37
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.3 kb
#include <cstdio>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> v[100010];
int logg[200010],l,rmq[200010][20],niv[100010],first[100010],vaz[100010];

void dfs(int nod,int nivel)
{
    vaz[nod]=1;
    rmq[++l][0]=nod;
    niv[nod]=nivel;
    first[nod]=l;
    for(int i=0;i<v[nod].size();i++)
        if(vaz[v[nod][i]]==0)
        {
            dfs(v[nod][i],nivel+1);
            rmq[++l][0]=nod;
        }
}

int lca(int a,int b)
{
    a=first[a];
    b=first[b];
    if(a>b) swap(a,b);
    int l=logg[b-a+1];
    if(niv[rmq[a][l]]<niv[rmq[b-(1<<l)+1][l]]) return rmq[a][l];
    else return rmq[b-(1<<l)+1][l];
}

int main()
{
    freopen("lca.in","r",stdin);
    freopen("lca.out","w",stdout);
    int n,m,sol,l1,x,a,b;
    scanf("%d%d",&n,&m);
    for(int i=2;i<=n;i++)
    {
        scanf("%d",&x);
        v[x].push_back(i);
    }
    dfs(1,0);
    logg[1]=0;
    for(int i=2;i<=l;i++)
        logg[i]=logg[i/2]+1;
    for(int i=1;i<=logg[l];i++)
        for(int j=1;j+(1<<i)<=l+1;j++)
            if(niv[rmq[j][i-1]]<=niv[rmq[j+(1<<(i-1))][i-1]]) rmq[j][i]=rmq[j][i-1];
            else rmq[j][i]=rmq[j+(1<<(i-1))][i-1];
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&a,&b);
        printf("%d\n",lca(a,b));
    }
    return 0;
}