Cod sursa(job #1771444)

Utilizator PaulStighiStiegelbauer Paul-Alexandru PaulStighi Data 5 octombrie 2016 17:22:04
Problema Lowest Common Ancestor Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 0.99 kb
#include <fstream>
#include <vector>
using namespace std;

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

const int NMax = 100005;
int N,Q,TT[NMax],Level[NMax];
vector<int> G[NMax];
void Read()
{
    fin>>N>>Q;
    for(int i = 2; i <= N; ++i)
        {
            fin>>TT[i];
            G[TT[i]].push_back(i);
        }
}

void DFS(int Nod)
{
    for(int i = 0; i < (int)G[Nod].size(); ++i)
        {
            int Vecin = G[Nod][i];
            Level[Vecin] = Level[Nod] + 1;
            DFS(Vecin);
        }
}

int LCA(int x,int y)
{
    if(Level[x]<Level[y])
            swap(x,y);

        while(Level[x] != Level[y])
                x = TT[x];

        while(x!=y)
        {
            x=TT[x];
            y=TT[y];
        }

    return x;
}

void SolveandPrint()
{
    DFS(1);
    while(Q--)
    {
        int x,y;
        fin>>x>>y;

        fout<<LCA(x,y)<<"\n";
    }
}

int main()
{
    Read();
    SolveandPrint();
    return 0;
}