Cod sursa(job #2565982)

Utilizator stefan_creastaStefan Creasta stefan_creasta Data 2 martie 2020 18:12:38
Problema Lowest Common Ancestor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.41 kb
#include <cstdio>
#include <vector>
#include <algorithm>
#include <ctype.h>
using namespace std;

/** Funcţiile necesare parsării fişierului de intrare **/
FILE *_fin;
int _in_loc; char _in_buff[4096];


void read_init(const char* nume) // Apelaţi această funcţie la începutul funcţiei <main>
{
	_fin = fopen(nume, "r");
	_in_loc = 4095;
}

char read_ch()
{
	++_in_loc;
	if (_in_loc == 4096) {
		_in_loc = 0;
		fread(_in_buff, 1, 4096, _fin);
	}
	return _in_buff[_in_loc];
}

int read_u32() // Apelaţi această funcţie pentru a citi un număr ce se încadrează în categoria <unsigned int>
{
	int u32 = 0; char c;
	while (!isdigit(c = read_ch()) && c != '-');
	int sgn = 1;
	if (c == '-') {
		sgn = -1;
	} else {
		u32 = c - '0';
	}
	while (isdigit(c = read_ch())) {
		u32 = u32 * 10 + c - '0';
	}
	return u32 * sgn;
}
const int NMAX = 2 * 100015;
vector <int> vec[NMAX];
int h[NMAX];
int poz[NMAX];
int top = 0;
int rmq[NMAX][30];
int lg[NMAX];
void dfs(int u, int f) {
    h[u] = h[f] + 1;
    rmq[++top][0] = u;
    poz[u] = top;
    for(int v : vec[u]) {
        if(v != f) {
            rmq[++top][0] = u;
            poz[u] = top;
            dfs(v, u);
        }
    }
}

int main() {
    int n, m, i, j, l, f, a, b, dist, poza, pozb, n1, n2;
    read_init("lca.in");
    freopen("lca.out", "w", stdout);
    n = read_u32();
    m = read_u32();
    for(i = 2; i <= n; i++) {
        f = read_u32();
        vec[i].push_back(f);
        vec[f].push_back(i);
    }
    dfs(1, 0);
    for(int i = 2; i <= top; i++) {
        lg[i] = lg[i >> 1] + 1;
    }
    for(l = 1, j = 1; l <= top; j++, l <<= 1) {
        for(i = 1; i <= top - l; i++) {
            if(h[rmq[i][j - 1]] < h[rmq[i + l][j - 1]]) {
                rmq[i][j] = rmq[i][j - 1];
            }
            else {
                rmq[i][j] = rmq[i + l][j - 1];
            }
        }
    }
    for(i = 1; i <= m; i++) {
        int a, b;
        a = read_u32();
        b = read_u32();
        poza = min(poz[a], poz[b]);
        pozb = max(poz[a], poz[b]);
        dist = pozb - poza + 1;
        n1 = h[rmq[poza][lg[dist]]];
        n2 = h[rmq[pozb - (1 << lg[dist]) + 1][lg[dist]]];
        if(n1 < n2) {
            printf("%d\n", rmq[poza][lg[dist]]);
        }
        else {
            printf("%d\n", rmq[pozb - (1 << lg[dist]) + 1][lg[dist]]);
        }
    }
    return 0;
}