Cod sursa(job #1075910)

Utilizator Iustin_BulimarFMI Iustin Bulimar Iustin_Bulimar Data 9 ianuarie 2014 18:33:29
Problema Lowest Common Ancestor Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.41 kb
#include <fstream>
#include <vector>
#include <cmath>
using namespace std;
ifstream cin("lca.in");
ofstream cout("lca.out");

const int n_max=100001;
const int log_n_max=17;

vector<int> a[n_max];
int n, q, i, x, y, k, niv, nivel[n_max], e[2*n_max+4], p[n_max], m[n_max][log_n_max];
bool viz[n_max];

void mark(int nod)
{
    e[++k]=nod;
    if(!p[nod]) p[nod]=k;
}

void dfs(int nod)
{
    int i;
    viz[nod]=1;
    mark(nod);
    for(i=0; i<a[nod].size(); i++)
        if(!viz[a[nod][i]])
        {
            dfs(a[nod][i]);
            mark(nod);
        }
}

void rmq(int v[], int n)
{
    int i, j;
    for(i=0; i<n; i++) m[i][0]=i;
    for(j=1; 1<<j<=n; j++)
        for(i=0; i+(1<<j)-1<n; i++)
            if(nivel[v[m[i][j-1]]]<nivel[v[m[i+(1<<(j-1))][j-1]]]) m[i][j]=m[i][j-1];
            else m[i][j]=m[i+(1<<(j-1))][j-1];
}

int query(int v[], int x, int y)
{
    int k=log2(y-x+1);
    if(nivel[v[m[x-1][k]]]<nivel[v[m[y-(1<<k)][k]]]) return m[x-1][k];
    return m[y-(1<<k)][k];
}

int main()
{
    cin>>n>>q;
    nivel[1]=1;
    for(i=2; i<=n; i++)
    {
        cin>>x;
        a[x].push_back(i);
        nivel[i]=nivel[x]+1;
    }
    dfs(1);
    //for(i=1; i<=k; i++) cout<<e[i]<<" "<<p[i]<<endl;
    rmq(e, k);
    while(q--)
    {
        cin>>x>>y;
        if(p[x]>p[y]) swap(x, y);
        cout<<e[query(e ,p[x], p[y])]<<'\n';
    }
    return 0;
}