Cod sursa(job #1909437)

Utilizator trutruvasilicaHuhurez Marius trutruvasilica Data 7 martie 2017 12:43:46
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.27 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("lca.in");
ofstream fout("lca.out");
const int N=100005;
int lev[50*N],val[50*N],apar[N],dp[25][50*N],lg[50*N],k;
vector<int>H[N];
void dfs(int nod,int lvl)
{
    k++;
    lev[k]=lvl;
    apar[nod]=k;
    val[k]=nod;
    for(vector<int>::iterator it=H[nod].begin();it!=H[nod].end();it++)
    {
        dfs(*it,lvl+1);
        k++;
        lev[k]=lvl;
        val[k]=nod;
    }
}
void rmq()
{
    int i,j;
    dp[0][1]=val[1];
    for(i=2;i<=k;i++)
    {
        lg[i]=lg[i/2]+1;
        dp[0][i]=i;
    }
    for(i=1;(1<<i)<=k;i++)
    {
        for(j=1;j<=k-(1<<i);j++)
        {
            dp[i][j]=dp[i-1][j];
            if(lev[dp[i-1][j+(1<<(i-1))]]<lev[dp[i-1][j]]) dp[i][j]=dp[i-1][j+(1<<(i-1))];
        }
    }
}
int lca(int x,int y)
{
    int k,sol;
    int a=apar[x],b=apar[y];
    if(a>b) swap(a,b);
    k=lg[b-a];
    sol=dp[k][a];
    if(lev[dp[k][b-(1<<k)+1]]<lev[sol]) sol=dp[k][b-(1<<k)+1];
    return val[sol];
}
int main()
{
    int n,m,i,a,b;
    fin>>n>>m;
    for(i=2;i<=n;i++)
    {
        fin>>a;
        H[a].push_back(i);
    }
    dfs(1,0);
    rmq();
    for(i=1;i<=m;i++)
    {
        fin>>a>>b;
        fout<<lca(a,b)<<"\n";
    }
}