Cod sursa(job #754631)

Utilizator visanrVisan Radu visanr Data 2 iunie 2012 19:16:51
Problema Lowest Common Ancestor Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.73 kb
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <vector>
using namespace std;

#define nmax 100010

int n, m, k, Euler[3 * nmax], M[nmax][17], First[nmax], L[3 * nmax];
vector<int> G[nmax];

void dfs(int node, int level)
{
     Euler[++k] = node;
     L[k] = level;
     First[node] = k;
     for(int i = 0; i < G[node].size(); i++)
     {
                dfs(G[node][i], level + 1);
                Euler[++k] = node;
                L[k] = level;
     }
}


void process()
{
     int i, j;
     for(i = 0; i < k; i++) M[i][0] = i;
     for(j = 1; (1 << j) <= k; j++)
     {
           for(i = 0; i + (1 << j) - 1 < k; i++)
           {
                 if(L[M[i][j - 1]] < L[M[i + (1 << (j - 1))][j - 1]])
                 {
                             M[i][j] = M[i][j - 1];
                 }else
                 {
                      M[i][j] = M[i + (1 << (j - 1))][j - 1];
                 }
           }
     }
}


int LCA(int i, int j)
{
    int a, b, kk;
    a = First[i];
    b = First[j];
    if(a > b)
    {
         int aux = a;
         a = b;
         b = aux;
    }
    kk = (int) log2(b - a + 1);
    if(L[M[a][kk]] <= L[M[b - (1 << kk) + 1][k]]) return Euler[M[a][kk]];
    else return Euler[M[b - (1 << kk) + 1][kk]];
}

int main()
{
    freopen("lca.in", "r", stdin);
    freopen("lca.out", "w", stdout);
    int i, x, y;
    scanf("%i %i", &n, &m);
    k = -1;
    for(i = 2; i <= n; i++) 
    {
          scanf("%i", &x);
          G[x].push_back(i);
    }
    dfs(1, 0);
    k++;
    process();
    for(i = 0; i < m; i++)
    {
          scanf("%i %i", &x, &y);
          printf("%i\n", LCA(x, y));
    }
    scanf("%i", &i);
    return 0;
}