Pagini recente » Cod sursa (job #326526) | Cod sursa (job #2431527) | Cod sursa (job #1395139) | Cod sursa (job #755542) | Cod sursa (job #1944058)
#include <iostream>
#include <vector>
#include <fstream>
using namespace std;
ifstream f("lca.in");
ofstream g("lca.out");
vector <int> G[10005];
int n,m,k,first[10005],stk[10005 << 1],nivel[10005 << 1], lg[10005 << 1], rmq[25][10005 << 1];
void parc_euler(int x, int level)
{
stk[++k]=x;
nivel[x]=level;
first[x]=k;
for(int i=0;i<G[x].size();i++)
{
int niv=nivel[k];
parc_euler(G[x][i],level+1);
nivel[++k]=niv;
stk[k]=x;
}
}
int lca(int x, int y)
{
x=first[x];
y=first[y];
if(x>y) swap(x,y);
int l=lg[y-x+1];
int minim = rmq[l][x];
if(nivel[minim] > nivel[rmq[l][y- (1 << l)+1]]);
minim=rmq[l][y- (1 << l)+1];
return stk[minim];
}
int main()
{
f>>n>>m;
for(int i=2;i<=n;i++)
{
int x;
f>>x;
G[x].push_back(i);
}
parc_euler(1,0);
/*for(int i=1;i<=k;i++)
cout<<stk[i]<<' ';
cout<<'\n';
for(int i=1;i<=k;i++)
cout<<nivel[i]<<' ';*/
for(int i=2;i<=k;i++)
lg[i]=lg[i >> 1]+1;
for(int i=1;i<=k;i++)
rmq[0][i]=i;
for(int i=1;(1 << i)<=k;i++)
for(int j=1; j + (1 << i) -1 <=k;j++)
{
rmq[i][j]=rmq[i-1][j];
int nr= 1 << (i-1);
if(nivel[rmq[i][j]] > nivel[rmq[i-1][j+ nr]])
rmq[i][j]=rmq[i-1][j+ nr];
}
for(int i=1;i<=m;i++)
{
int a,b;
f>>a>>b;
g<<lca(a,b)<<'\n';
}
f.close();
g.close();
return 0;
}