Cod sursa(job #2025986)

Utilizator MihaelaCismaruMihaela Cismaru MihaelaCismaru Data 23 septembrie 2017 15:34:11
Problema Lowest Common Ancestor Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.48 kb
#include<fstream>
#include<bitset>
#include<vector>
using namespace std;
ifstream in("lca.in");
ofstream out("lca.out");
int rmq[29][200001],k,level[100001],p[30],euler[200001],n,q,x,y,a,b,aux[200001], prim[100001];
vector<int>v[100001];
bitset<200001>hz;
void dfs( int nod ){
    rmq[0][++k] = nod;
    if( v[nod].size() == 0 ){
        return;
    }
    else{
        for( int i = 0; i < v[nod].size(); i ++ ){
            dfs( v[nod][i] );
            rmq[0][++k] = nod;
        }
    }
    return;
}
int main(){
    in >> n >> q;
    level[1] = 1;
    for( int i = 2; i <= n; i ++ ){
        in >> x;
        v[x].push_back(i);
        level[i] = level[x]+1;
    }

    dfs( 1 );
    p[0] = 1;
    for( int i = 1; i <= 25; i ++ ){
        p[i] = p[i-1]*2;
        if( k >= p[i] ){
            a = i;
        }
    }
    for( int i = 2; i <= k; i ++ ){
        aux[i] = aux[i/2]+1;
    }
    for( int j = 1; j <= a; j ++ ){
        for( int i = 1; i <= k-p[j]+1; i ++ ){
            rmq[j][i] = min(rmq[j-1][i+p[j-1]],rmq[j-1][i] );
        }
    }
    for( int i = 1; i <= k; i ++ ){
        if( hz[rmq[0][i]] == 0 ){
            hz[rmq[0][i]] = 1;
            prim[ rmq[0][i] ] = i;
        }
    }
    for( int i = 1; i <= q; i ++ ){
        in >> a  >> b;
        a = min(prim[a],prim[b]);
        b = max(prim[a],prim[b]);
        y = aux[b-a+1];
        x = p[y];
        out<< min( rmq[y][b-x+1] ,rmq[y][a] )<<"\n";
    }
    return 0;
}