Cod sursa(job #2721034)

Utilizator victorzarzuZarzu Victor victorzarzu Data 11 martie 2021 15:15:40
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 kb
#include <bits/stdc++.h>
 
using namespace std;
ifstream f("lca.in");
ofstream g("lca.out");
int n, k, x, y, m[21][100005], nivel[100005], maxim;
vector<int> graf[100005];
 
void Read()
{
  f>>n>>k;
  for(int i = 2;i <= n;++i)
    f>>x, m[0][i] = x, graf[x].push_back(i);
  for(int j = 1;(1 << j) <= n;++j)
    for(int i = 1;i <= n;++i)
      m[j][i] = m[j - 1][m[j - 1][i]];
  for(maxim = 1;(1 << maxim) <= n;++maxim);
}
 
void dfs(int nod, int lvl)
{
  nivel[nod] = lvl;
  for(auto it : graf[nod])
    dfs(it, lvl + 1);
}
 
int lca(int x, int y)
{
  if(nivel[x] < nivel[y])
    swap(x, y);
 
  int diff = nivel[x] - nivel[y];
  for(int i = 0;(1 << i) <= diff;++i)
    if(diff & (1 << i))
      x = m[i][x];
 
  if(x == y) return x;
 
  for(int i = maxim;i >= 0;--i)
    if(m[i][x] && m[i][x] != m[i][y])
      x = m[i][x], y = m[i][y];
  
  return m[0][x];
}
 
void Solve()
{
  dfs(1, 0);
  for(int i = 1;i <= k;++i)
    f>>x>>y, g<<lca(x, y)<<'\n';
}
 
int main()
{
  Read();
  Solve();
  return 0;
}