Pagini recente » Cod sursa (job #1019038) | Cod sursa (job #3125842) | Cod sursa (job #2303776) | Cod sursa (job #1167628) | Cod sursa (job #1383925)
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("lca.in");
ofstream fout("lca.out");
int N, M;
vector<int> V[100005];
int K;
int EULER[200005], LVL[200005], First[100005], logaritm[100005], RMQ[20][400005];
void dfs(int nod, int nivel)
{
EULER[++K] = nod; //nodul actual este adaugat in reprezentarea Euler a arborelui
LVL[K] = nivel; //se retine nivelul fiecarei pozitii din reprezentarea Euler a arborelui
First[nod] = K; //se retine prima aparitie a fiecarui nod in reprezentarea Euler
for (vector<int>::iterator it = V[nod].begin(); it != V[nod].end(); ++it)
{
dfs(*it, nivel + 1);
EULER[++K] = nod;
LVL[K] = nivel;
}
}
void rmq()
{
//in RMQ[i][j] va fi nodul de nivel minim dintre pozitiile (j, j + 2 ^ (i - 1)) din reprezentarea Euler a arborelui
logaritm[1] = 0;
for (int i = 2; i <= K; ++i)
logaritm[i] = logaritm[i >> 1] + 1;
for (int i = 1; i <= K; ++i)
RMQ[0][i] = i;
for (int j = 1; (1 << j) <= K; ++j)
for (int i = 1; i + (1 << (j - 1)) <= K; ++i)
{
int p = 1 << (j - 1);
if (LVL[RMQ[j - 1][i]] <= LVL[RMQ[j - 1][i + p]]) RMQ[j][i] = RMQ[j - 1][i];
else RMQ[j][i] = RMQ[j - 1][i + p];
}
}
int lca(int nod1, int nod2)
{
//LCA-ul nodurilor nod1 si nod2 va fi nodul cu nivel minim din secventa (First[nod1], First[nod2]) din reprezentarea Euler
int sol;
int start = First[nod1];
int finish = First[nod2];
if (start > finish)
swap(start, finish);
int dif = finish - start + 1;
int l = logaritm[dif];
if (LVL[RMQ[l][start]] <= LVL[RMQ[l][finish - (1 << l) + 1]]) sol = RMQ[l][start];
else sol = RMQ[l][finish - (1 << l) + 1];
return EULER[sol];
}
int main()
{
fin >> N >> M;
for (int i = 2, nod; i <= N; ++i)
{
fin >> nod;
V[nod].push_back(i);
}
dfs(1, 0);
rmq();
for (int i = 1, nod1, nod2; i <= M; ++i)
{
fin >> nod1 >> nod2;
fout << lca(nod1, nod2) << '\n';
}
fin.close();
fout.close();
return 0;
}