Pagini recente » Cod sursa (job #1520918) | Cod sursa (job #1524972) | Cod sursa (job #2973238)
#include <iostream>
#include <fstream>
#include <vector>
#include <cstring>
#include <cmath>
#define int long long
#define NMAX 100002
#define LGMAX 20
using namespace std;
int n,x,y,m,euler[NMAX*2],k,niv[NMAX],rmq[LGMAX][NMAX],mbit[NMAX*2];
vector<int> v[NMAX];
pair<int, int> ap[NMAX];
ifstream fin("lca.in");
ofstream fout("lca.out");
void dfs(int nod, int tata){
euler[++k] = nod;
ap[nod] = {k, k};
for(int vecin: v[nod]){
if(niv[vecin] == 0){
niv[vecin] = 1+niv[nod];
dfs(vecin, nod);
euler[++k] = nod;
ap[nod].second = k;
}
}
}
int lca(int a, int b){
int st = ap[a].first;
int dr = ap[b].second;
if(st > dr){
swap(a, b);
st = ap[a].first;
dr = ap[b].second;
}
int len = dr-st+1;
int log = log2(len);
int x = rmq[log][st], y = rmq[log][dr-(1<<log)+1];
if(niv[x] < niv[y]){
return x;
}
return y;
}
signed main()
{
fin >> n >> m;
for(int i = 2; i <= n; i++){
fin >> x;
v[x].push_back(i);
v[i].push_back(x);
}
niv[1] = 1;
dfs(1, -1);
for(int i = 1; i <= k; i++){
rmq[0][i] = euler[i];
}
int lg = log2(k);
for(int i = 1; i <= lg; i++){
for(int j = 1; j+(1<<(i-1)) <= k; j++){
rmq[i][j] = rmq[i-1][j];
if(niv[rmq[i-1][j+(1<<(i-1))]] < niv[rmq[i][j]]){
rmq[i][j] = rmq[i-1][j+(1<<(i-1))];
}
}
}
while(m--){
fin >> x >> y;
fout << lca(x, y) << "\n";
}
return 0;
}