Cod sursa(job #1888535)

Utilizator mateigabriel99Matei Gabriel mateigabriel99 Data 22 februarie 2017 10:30:19
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.94 kb
#include <bits/stdc++.h>

#define NMax 100005
#define H 200

using namespace std;

ifstream fin("lca.in");
ofstream fout("lca.out");

int N,M;
int T[NMax],T2[NMax],niv[NMax];
vector<int> Graf[NMax];

void DFS(int nod,int t2,int nivel)
{
    niv[nod]=nivel;
    T2[nod]=t2;
    if(nivel%H==0)
        t2=nod;
    for(vector<int>::iterator it=Graf[nod].begin();it!=Graf[nod].end();it++)
        DFS(*it,t2,nivel+1);
}

int LCA(int x,int y)
{
    while(T2[x]!=T2[y])
        if(niv[x]>niv[y])
            x=T2[x];
        else
            y=T2[y];

    while(x!=y)
        if(niv[x]>niv[y])
            x=T[x];
        else
            y=T[y];

    return x;
}

int main()
{
    fin>>N>>M;
    for(int i=2;i<=N;i++)
    {
        fin>>T[i];
        Graf[T[i]].push_back(i);
    }
    DFS(1,0,0);
    while(M--)
    {
        int x,y;
        fin>>x>>y;
        fout<<LCA(x,y)<<"\n";
    }

    return 0;
}