Cod sursa(job #1744898)

Utilizator Dan_RadulescuRadulescu Dan Dan_Radulescu Data 20 august 2016 18:13:04
Problema Lowest Common Ancestor Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.67 kb
#include<fstream>
#include<stdlib.h>
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];
struct nod2{
   int inf;
   nod2 *urm;
}*L[100005];
void adaugsf(nod2 *&vf,int val){
    nod2 *q;
    q=new nod2;
    q->inf=val;
    q->urm=vf;
    vf=q;
}
void cit(){
    fin>>n>>m;
    for (i=1;i<=n;i++)
        L[i]=0;
    for (i=2;i<=n;i++){
        fin>>val;
        adaugsf(L[val],i);
    }
}
void DFS(int k,int niv){
    nod2 *q;
    nr++;
    nivel[nr]=niv;
    nod[nr]=k;
    ap[k]=nr;
    q=L[k];
    while(q!=0){
        DFS(q->inf,niv+1);
        nr++;
        nivel[nr]=niv;
        nod[nr]=k;
        q=q->urm;
    }
}
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;
}