Pagini recente » Cod sursa (job #2089899) | Cod sursa (job #589029) | Cod sursa (job #194818) | Cod sursa (job #2711547) | Cod sursa (job #3307287)
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
using namespace std;
ifstream fin("lca.in");
ofstream fout("lca.out");
const int MAX=1e5,MAXP=16;
int n,i,t[MAX+5],depth[MAX+5],node1,node2,q,v[2*MAX+5],nr,p_end[MAX+5],p_start[MAX+5],r[MAXP+5][MAX+5];
bool viz[MAX+5];
vector <int> edge[MAX+5];
bool is_ancestor(int x, int y)
{
return p_start[x]<=p_start[y] && p_end[y]<=p_end[x];
}
void dfs(int node)
{
v[++nr]=node;
viz[node]=1;
for (auto node2: edge[node])
{
if (!viz[node2])
{
depth[node2]=depth[node]+1;
dfs(node2);
}
}
v[++nr]=node;
}
int lca(int x, int y)
{
if (is_ancestor(x,y))
return x;
if (is_ancestor(y,x))
return y;
for (int p=MAXP; p>=0; p--)
{
int node=r[p][x];
if (node!=0 && !is_ancestor(node,y))
x=node;
}
return r[0][x];
}
void precalc()
{
for (i=1; i<=n; i++)
r[0][i]=t[i];
int p;
for (p=1; (1<<p)<=n; p++)
{
for (i=1; i<=n; i++)
{
int node=r[p-1][i];
r[p][i]=r[p-1][node];
}
}
}
int main()
{
ios_base::sync_with_stdio(false);
fin.tie(0); fout.tie(0);
fin>>n>>q;
for (i=2; i<=n; i++)
{
fin>>t[i];
edge[i].push_back(t[i]);
edge[t[i]].push_back(i);
}
dfs(1);
for (i=1; i<=nr; i++)
p_end[v[i]]=i;
for (i=nr; i>=1; i--)
p_start[v[i]]=i;
precalc();
while (q--)
{
fin>>node1>>node2;
fout<<lca(node1,node2)<<'\n';
}
return 0;
}