Cod sursa(job #1218310)

Utilizator MaarcellKurt Godel Maarcell Data 10 august 2014 14:04:41
Problema Lowest Common Ancestor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.53 kb
#include <fstream>
#include <vector>
#include <iostream>
#define MAXN 100050
using namespace std;

int N,M,K,ql,qr,MIN,anc, first[MAXN*2], lvl[MAXN*2], NOD[MAXN*2], tree[MAXN*4]; vector<int> graf[MAXN];

void build_tree(int nod, int l, int r){
    if (l==r){
        tree[nod]=l;
        return;
    }

    int mid=(l+r)/2;
    build_tree(nod*2,l,mid);
    build_tree(nod*2+1,mid+1,r);
    if (lvl[tree[nod*2]]<lvl[tree[nod*2+1]])
        tree[nod]=tree[nod*2];
    else
        tree[nod]=tree[nod*2+1];
}

void query(int nod, int l, int r){
    if (ql<=l && r<=qr){
        if (lvl[tree[nod]]<MIN){
            MIN=lvl[tree[nod]];
            anc=NOD[tree[nod]];
        }
            return;
    }

    int mid=(l+r)/2;
    if (ql<=mid) query(nod*2,l,mid);
    if (mid<qr)  query(nod*2+1,mid+1,r);
}

void dfs(int nod, int level){
    NOD[++K]=nod;
    lvl[K]=level;
    first[nod]=K;

    for (int i=0; i<graf[nod].size(); i++){
        dfs(graf[nod][i],level+1);
        NOD[++K]=nod;
        lvl[K]=level;
    }
}

int lca(int a, int b){
    ql=first[a]; qr=first[b];
    if (qr<ql)
        swap(qr,ql);

    MIN=(1<<29);
    query(1,1,K);

    return anc;
}
int main(){
    ifstream in("lca.in");
    ofstream out("lca.out");
    in >> N >> M;

    int i,x,y;
    for (i=2; i<=N; i++){
        in >> x;
        graf[x].push_back(i);
    }

    dfs(1,0);
    build_tree(1,1,K);

    for (i=1; i<=M; i++){
        in >> x >> y;
        out << lca(x,y) << "\n";
    }
    return 0;
}