Cod sursa(job #1464720)

Utilizator fanache99Constantin-Buliga Stefan fanache99 Data 24 iulie 2015 12:44:03
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.33 kb
#include<cstdio>
#include<vector>
using namespace std;
int nr=0,depth[200010],euler[200010],log[200010],first[100010],rmq[200010][25];
vector<int> v[100010];
void dfs(int nod, int h){
    int i,dim=v[nod].size();
    nr++;
    euler[nr]=nod;
    depth[nr]=h;
    first[nod]=nr;
    for(i=0;i<dim;i++){
        dfs(v[nod][i],h+1);
        nr++;
        euler[nr]=nod;
        depth[nr]=h;
    }
}
int lca(int x,int y){
    int sol,aux;
    if(x>y){
        aux=x;
        x=y;
        y=aux;
    }
    sol=rmq[x][log[y-x+1]];
    if(depth[sol]>depth[rmq[y+1-(1<<log[y-x+1])][log[y-x+1]]])
        sol=rmq[y+1-(1<<log[y-x+1])][log[y-x+1]];
    printf("%d\n",euler[sol]);
}
int main(){
    freopen("lca.in","r",stdin);
    freopen("lca.out","w",stdout);
    int n,m,x,y,i,j;
    scanf("%d%d",&n,&m);
    for(i=2;i<=n;i++){
        scanf("%d",&x);
        v[x].push_back(i);
    }
    dfs(1,0);
    for(i=1;i<=nr;i++)
        rmq[i][0]=i;
    for(j=1;(1<<j)<nr;j++)
        for(i=1;i+(1<<j)<=nr;i++){
            rmq[i][j]=rmq[i][j-1];
            if(depth[rmq[i+(1<<(j-1))][j-1]]<depth[rmq[i][j]])
                rmq[i][j]=rmq[i+(1<<(j-1))][j-1];
        }
    for(i=2;i<=nr;i++)
        log[i]=log[i/2]+1;
    for(i=1;i<=m;i++){
        scanf("%d%d",&x,&y);
        lca(first[x],first[y]);
    }
    return 0;
}