Pagini recente » Cod sursa (job #1630378) | Cod sursa (job #3129744) | Cod sursa (job #1640582) | Cod sursa (job #2642384) | Cod sursa (job #2492353)
#include <bits/stdc++.h>
using namespace std;
vector <int> tour;
vector <int> v[100001];
bitset <100001> viz;
int lvl[100001];
int n;
int rmq[18][100001],logs[100001];
map <int,int> mp;
void DFS(int nod,int level){
lvl[nod] = level;
tour.push_back(nod);
mp[nod] = tour.size() - 1;
viz[nod] = 1;
for(auto x : v[nod]){
if(!viz[x]){
DFS(x,level + 1);
tour.push_back(nod);
mp[nod] = tour.size() - 1;
}
}
}
void compute(){
int i,j,lung;
for(i = 1;i < tour.size();i++){
rmq[0][i] = tour[i];
}
for(i = 2;i <= tour.size() - 1;i++){
logs[i] = logs[i/2] + 1;
}
for(i = 1;i <= logs[tour.size() - 1];i++){
int lung = 1 << i;
for(j = 1;j + lung - 1 <= tour.size() - 1;j++){
int a = rmq[i-1][j];
int b = rmq[i-1][j + lung/2];
if(lvl[a] > lvl[b])
rmq[i][j] = b;
else
rmq[i][j] = a;
}
}
}
int getmin(int a,int b){
a = mp[a];
b = mp[b];
if(b < a)
swap(a,b);
int cv = logs[b - a + 1];
int lung = 1 << cv;
int x = rmq[cv][a];
int y = rmq[cv][b - lung + 1];
if(lvl[x] > lvl[y])
return y;
return x;
}
int main()
{
ifstream cin("lca.in");
ofstream cout("lca.out");
int m,i,x;
cin >> n >> m;
tour.push_back(0);
for(i = 2;i <= n;i++){
cin >> x;
if(x != i){
v[x].push_back(i);
v[i].push_back(x);
}
}
for(i = 1;i <= n;i++){
if(!viz[i]){
DFS(i,1);
}
} compute();
//for(auto x : tour)
// cout << x << " ";
while(m--){
int y;
cin >> x >> y;
cout << getmin(x,y) << "\n";
}
return 0;
}