Pagini recente » Cod sursa (job #2301285) | Cod sursa (job #2743333) | Cod sursa (job #2603868) | Cod sursa (job #2173396) | Cod sursa (job #374616)
Cod sursa(job #374616)
#include <fstream>
#include <vector>
#define MaxN 100528
#define MaxL 100
using namespace std;
fstream fin ("lca.in",ios::in);
fstream fout("lca.out",ios::out);
struct nod {
int x;
nod *urm;
};
int n, m, viz[MaxN], k, parc[MaxN<<2], dep[MaxN<<2];
nod *L[MaxN];
int first[MaxN], Lg[MaxN<<2 + 3], M[MaxL][MaxN];
void rmq(){
for(int i = 2; i <= k; ++i)
Lg[i] = Lg[i >> 1] + 1;
for(int i = 1; i <= k; ++i)
M[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);
M[i][j] = M[i-1][j];
if(dep[M[i-1][j + l]] < dep[M[i][j]])
M[i][j] = M[i-1][j + l];
}
};
void add_nod(int ex1, int ex2){
nod *p = new nod;
nod *q = new nod;
p->x = ex2;
p->urm = L[ex1];
L[ex1] = p;
q->x = ex1;
q->urm = L[ex2];
L[ex2] = q;
};
void euler_det(int x, int depth){ //dfs euler
viz[x] = 1;
parc[++k] = x;
dep[k] = depth;
first[x] = k;
for (nod *p = L[x]; p != NULL; p = p->urm)
if (!viz[p->x]){
euler_det(p->x, depth + 1);
parc[++k] = x;
dep[k] = depth;
};
};
int lca(int x, int y)
{
//LCA-ul nodurilor x si y va fi nodul cu nivel minim din secventa (First[x], First[y]) din reprezentarea Euler a arborelui
int diff, l, sol, sh;
int a = first[x], b = first[y];
if(a > b) swap(a, b);
diff = b - a + 1;
l = Lg[diff];
sol = M[l][a];
sh = diff - (1 << l);
if(dep[sol] > dep[M[l][a + sh]])
sol = M[l][a + sh];
return parc[sol];
}
int main(){
int x, x1, x2;
fin>>n>>m;
for (int i = 2; i <= n; ++i){
fin>>x;
add_nod(i, x);
};
euler_det(1, 0);
//preprocesarea RMQ
Lg[1] = 0;
for (int i = 2; i <= k; i++)
Lg[i] = Lg[i >> 1] + 1;
rmq();
//efectuarea query-urilor
/*for (int i = 1; i <= k; ++i)
fout<<parc[i]<<' ';
fout<<'\n';
for (int i = 1; i <= k; ++i)
fout<<first[i]<<' ';
fout<<'\n';*/
for (int i = 1; i <= m; ++i){
fin>>x1>>x2;
fout<<lca(x1,x2)<<'\n';
};
return 0;
};