Cod sursa(job #1678336)

Utilizator GinguIonutGinguIonut GinguIonut Data 7 aprilie 2016 10:59:17
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.65 kb
#include <fstream>
#include <vector>

#define nMax 200009
#define pb push_back
#define mkp make_pair
#define pii pair<int, int>
#define lgMax 19
#define qMax 2000001
using namespace std;
ifstream fin("lca.in");
ofstream fout("lca.out");
int n, m, k;
int Lev[nMax/2], First[nMax/2], rmq[lgMax][nMax], lg[nMax];
vector<int> G[nMax];
pii q[qMax];
void read()
{
    int a, b;
    fin>>n>>m;

    for(int i=2;i<=n;i++)
    {
        fin>>a;
        G[a].pb(i);
    }

    for(int i=1;i<=m;i++)
    {
        fin>>a>>b;
        q[i]=mkp(a, b);
    }
}
void dfs(int node)
{
    rmq[0][++k]=node;
    First[node]=k;

    for(vector<int>::iterator it=G[node].begin();it!=G[node].end();it++)
    {
        int fiu=*it;

        Lev[fiu]=Lev[node]+1;
        dfs(fiu);
        rmq[0][++k]=node;
    }
}
void range_minimum_query()
{
    for(int i=2;i<nMax;i++)
        lg[i]=lg[i/2]+1;

    for(int len=1;(1 << len)<=k;len++)
    {
        int diff=(1 << (len-1));
        for(int i=1;i+(1 << len)-1<=k;i++)
        {
            rmq[len][i]=rmq[len-1][i];
            if(Lev[rmq[len-1][i+diff]]<Lev[rmq[len][i]])
                rmq[len][i]=rmq[len-1][i+diff];
        }
    }
}
void query(int a, int b)
{
    if(a>b)
        swap(a, b);

    int l=lg[b-a+1];
    int diff=(1 << l);

    int Sol=rmq[l][a];
    if(Lev[Sol]>Lev[rmq[l][b-diff+1]])
        Sol=rmq[l][b-diff+1];

    fout<<Sol<<'\n';
}
void solve()
{
    Lev[1]=1;
    dfs(1);

    range_minimum_query();

    for(int i=1;i<=m;i++)
        query(First[q[i].first], First[q[i].second]);
}
int main()
{
    read();
    solve();
    return 0;
}