Cod sursa(job #1744904)

Utilizator Dan_RadulescuRadulescu Dan Dan_Radulescu Data 20 august 2016 18:23:53
Problema Lowest Common Ancestor Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.53 kb
#include<fstream>
#include<stdlib.h>
#include<vector>
using namespace std;
ifstream fin("lca.in");
ofstream fout("lca.out");
int n,m,i,j,nr,ap[200010],val,a,b,log[200010],minim[20][200010],mins,maxs,l,nod[200010],nivel[200010];
vector<int>L[100005];
void cit(){
    fin>>n>>m;
    for (i=2;i<=n;i++){
        fin>>val;
        L[val].push_back(i);
    }
}
void DFS(int k,int niv){
    nr++;
    nivel[nr]=niv;
    nod[nr]=k;
    ap[k]=nr;
    vector<int> ::iterator  it;
    for(it=L[k].begin();it<L[k].end();it++)
    {
        DFS(*it ,niv+1);
        nr++;
        nod[nr]=k;
        nivel[nr]=niv;
    }

}
int main(){
    cit();
    DFS(1,0);
    log[1]=0;
    for (i=2;i<=nr;i++){
        log[i]=log[i/2]+1;
        minim[0][i]=i;
    }
    for (j=1;(1<<j)<=nr;j++){
        i=1;l=1<<(j-1);
        while(i+2*l<=nr){
            if (nivel[minim[j-1][i]]<nivel[minim[j-1][i+l]]) minim[j][i]=minim[j-1][i];
                 else
                    minim[j][i]=minim[j-1][i+l];
            i++;
        }
    }
    for (i=1;i<=m;i++){
        fin>>a>>b;
        if (ap[a]<ap[b]){
            mins=ap[a];
            maxs=ap[b];
        }
           else
           {
               maxs=ap[a];
               mins=ap[b];
           }
        val=log[maxs-mins+1];l=1<<val;
        if (nivel[minim[val][mins]]<nivel[minim[val][maxs-l+1]]) fout<<nod[minim[val][mins]]<<'\n';
           else
            fout<<nod[minim[val][maxs-l+1]]<<'\n';
    }
    fin.close();
    fout.close();
    return 0;
}