Pagini recente » Cod sursa (job #1075924) | Cod sursa (job #706677) | Cod sursa (job #1344502) | Cod sursa (job #1023757) | Cod sursa (job #1537872)
#include <iostream>
#include <fstream>
#include <vector>
#define NMAX 100005
/// RMQ, DFS, parcurgere Euler
using namespace std;
ifstream f("lca.in");
ofstream g("lca.out");
vector <int> v[100005]; /// vectorul de tati
int n, m;
int level[NMAX], euler[2 * NMAX], first[NMAX], log[2 * NMAX];
int cnt;
int rmq[18][2 * NMAX]; /// minimul
int nod[18][2 * NMAX]; /// nodul
void dfs(int x)
{
/// cnt - contor
euler[++cnt] = x;
first[x] = cnt;
for (auto fiu:v[x])
{
level[fiu] = level[x] + 1;
dfs(fiu);
euler[++cnt] = x;
}
}
void make_log(void)
{
log[1] = 0;
for (int i = 2; i <= cnt; i++)
log[i] = log[i / 2] + 1;
}
int lca(int x, int y)
{
int p;
int diff;
x = first[x];
y = first[y];
if (x > y)
swap(x, y);
diff = y - x + 1;
p = log[diff];
if (rmq[p][x] < rmq[p][y - (1 << p) + 1])
return nod[p][x];
return nod[p][y - (1 << p) + 1];
}
void rezolvare()
{
int x;
int nod1, nod2;
f >> n >> m;
for (int i = 1; i <= n - 1; i++)
{
f >> x;
v[x].push_back(i + 1);
}
dfs(1);
make_log();
for (int i = 1; i <= cnt; i++)
{
rmq[0][i] = level[euler[i]];
nod[0][i] = euler[i];
}
for (int i = 1; (1 << i) <= cnt; i++)
{
for (int j = 1; j + (1 << (i - 1)) <= cnt; j++)
{
if (rmq[i - 1][j] < rmq[i - 1][j + (1 << (i - 1))])
{
rmq[i][j] = rmq[i - 1][j];
nod[i][j] = nod[i - 1][j];
}
else
{
rmq[i][j] = rmq[i - 1][j + (1 << (i - 1))];
nod[i][j] = nod[i - 1][j + (1 << (i - 1))];
}
}
}
for (int i = 1; i <= m; i++)
{
f >> nod1 >> nod2;
g << lca(nod1, nod2) << '\n';
}
}
int main()
{
rezolvare();
return 0;
}