Pagini recente » Cod sursa (job #2255159) | Cod sursa (job #411931) | Cod sursa (job #3149658) | Cod sursa (job #40833) | Cod sursa (job #3227914)
#include <fstream>
#define maxlog 19
#define dim 100001
#include <vector>
using namespace std;
ifstream fin("lca.in");
ofstream fout("lca.out");
struct rmqitem{
int nod, niv;
}rmq[maxlog][dim];
vector <int> l[dim];
int tata[dim], pstart[dim], bit[dim];
int n, pcrt;
void calculeaza_putere()
{
bit[1]=0;
for(int i=2;i<=n;i++)
{
bit[i]=1+bit[i/2];
}
}
void dfs(int nod, int niv)
{
pstart[nod]=++pcrt;
rmq[0][pcrt]={nod, niv};
for(int i=0;i<l[nod].size();i++)
{
dfs(l[nod][i], niv+1);
rmq[0][++pcrt]={nod, niv};
}
}
void calculeaza_rmq()
{
for(int p=1;p<maxlog;p++)
{
for(int i=1;i<=pcrt;i++)
{
rmq[p][i]=rmq[p-1][i];
int j=i+(1<<(p-1))-1;
if(j<=pcrt&&rmq[p-1][j].niv<rmq[p][i].niv)
{
rmq[p][i]=rmq[p-1][j];
}
}
}
}
int lca(int x, int y)
{
int px=pstart[x];
int py=pstart[y];
if(px>py)
{
swap(px, py);
}
int dime=py-px+1;
int e=bit[dime];
int len=(1<<e);
rmqitem st, dr;
st=rmq[e][px];
dr=rmq[e][py-len+1];
if(st.niv<dr.niv)
{
return st.nod;
}
else
{
return dr.nod;
}
}
int main()
{
int m, i, x, y;
fin>>n>>m;
tata[1]=0;
for(int i=2;i<=n;i++)
{
fin>>tata[i];
l[tata[i]].push_back(i);
}
pcrt=0;
dfs(1, 0);
calculeaza_rmq();
calculeaza_putere();
for(i=1;i<=m;i++)
{
fin>>x>>y;
fout<<lca(x, y)<<"\n";
}
}