Cod sursa(job #1413200)

Utilizator gapdanPopescu George gapdan Data 1 aprilie 2015 18:45:09
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.57 kb
#include <cstdio>
#include <vector>
#include <algorithm>
#define NMAX 100005

using namespace std;

int n,m,k,t;
int lv[NMAX*4],H[NMAX*4],lg[NMAX*4],first[NMAX];
vector<int>v[NMAX];
int rmq[30][NMAX*4];

void dfs(int nod,int niv)
{
    lv[++k]=niv;
    H[k]=nod;
    first[nod]=k;
    for(int i = 0; i < v[nod].size(); ++i)
        {
            dfs(v[nod][i],niv+1);
            H[++k]=nod;
            lv[k]=niv;
        }
}

void citire()
{
   freopen("lca.in","r",stdin);
   scanf("%d%d",&n,&m);
   for(int i = 2; i <= n; ++i)
   {
       scanf("%d",&t);
       v[t].push_back(i);
   }
   dfs(1,0);
}

void Rmq()
{
    lg[1]=0;
    for(int i = 2; i <= k; ++i)
        lg[i] = lg[i/2] + 1;
    for(int i = 1; i <=k; ++i )
        rmq[0][i] = i;

    for(int i = 1; (1<<i) < k; ++i)
        for(int j = 1; j <= k - (1<<i); ++j)
        {
          int l = 1 << (i-1);
          rmq[i][j] = rmq[i-1][j];
          if(lv[rmq[i][j]] > lv[rmq[i-1][l + j]])  rmq[i][j] = rmq[i-1][j+l];
        }
}

int lca(int x, int y)
{
    int st,dr,dif,l,sh;
     st = first[x];
     dr = first[y];
    if (st > dr) swap(st,dr);
     dif = (dr-st+1);
     l = lg[dif];
     sh = dif - (1<<l);
     int sol = rmq[l][st];
    if (lv[sol] > lv[rmq[l][st+sh]]) sol = rmq[l][st+sh];
    return H[sol];
}

void query()
{
    int x,y;
    freopen("lca.out","w",stdout);
    for(int i = 1; i <= m; ++i)
    {
        scanf("%d%d",&x,&y);
        printf("%d\n",lca(x,y));
    }
}

int main()
{
    citire();
    Rmq();
    query();
    return 0;
}