Cod sursa(job #2150444)

Utilizator usureluflorianUsurelu Florian-Robert usureluflorian Data 3 martie 2018 16:03:26
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.98 kb
#include <cstdio>
#include <vector>
using namespace std;
const int nmax=1e5+3;
const int h=200;
vector <int> v[nmax];
int lvl[nmax],t[nmax],n,m,a,b,t2[nmax];
void dfs(int nod,int ant,int lv)
{
      lvl[nod]=lv;
      t2[nod]=ant;
      if(lv%h==0) ant=nod;
      for(int i=0;i<v[nod].size();++i) dfs(v[nod][i],ant,lv+1);
}
int main()
{
      freopen("lca.in","r",stdin);
      freopen("lca.out","w",stdout);
      scanf("%d%d",&n,&m);
      for(int i=2;i<=n;++i)
      {
            scanf("%d",&t[i]);
            v[t[i]].push_back(i);
      }
      lvl[1]=1;
      dfs(1,0,0);
      while(m--)
      {
            scanf("%d%d",&a,&b);
            while(t2[a]!=t2[b])
            {
                  if(lvl[a]>lvl[b]) a=t2[a];
                  else b=t2[b];
            }
            while(a!=b)
            {
                  if(lvl[a]>lvl[b]) a=t[a];
                  else b=t[b];
            }
            printf("%d\n",a);
      }
      return 0;
}