Cod sursa(job #566201)

Utilizator bacilaBacila Emilian bacila Data 28 martie 2011 19:22:55
Problema Lowest Common Ancestor Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1 kb
#include<fstream>
#include<vector>
using namespace std;
int E[200005],H[200005],k,prim[100005],n,m,q[200005][18];
vector<int> v[100005];

void dfs(int nod,int niv)
{E[++k]=nod;
H[k]=niv;
prim[nod]=k;
     for(int i=0;i<v[nod].size();i++)
     {
     dfs(v[nod][i],niv+1);
                   E[++k]=nod;
                   H[k]=niv;       
     }       
     
}
inline int mini(int x,int y)
{if(H[x]>H[y])
return y;
return x;}
int main()
{ifstream f("lca.in");
int i,x,y,j;
    f>>n>>m;
    for(i=2;i<=n;i++)
    {f>>x;
    v[x].push_back(i);}
    dfs(1,0);
   
    for(i=1;i<=k;i++)
    q[i][0]=i;
    
    for(i=1;(1<<i)<=k;i++)
    for(j=1;j+(1<<(i))<=k;j++)
    {q[j][i]=mini(q[j+(1<<(i-1))][i-1],q[j][i-1]);
    }
     ofstream g("lca.out");
    
    
    for(i=1;i<=m;i++)
    {f>>x>>y;
    x=prim[x]; y=prim[y];
    if(x>y) swap(x,y);
    k=31-__builtin_clz(y-x+1);
    g<<E[min(q[x][k],q[y-(1<<k)+1][k])]<<'\n';
    }
    g.close();
    f.close();
return 0;
}