-
Notifications
You must be signed in to change notification settings - Fork 8
Expand file tree
/
Copy pathLCA2class.cpp
More file actions
94 lines (87 loc) · 1.75 KB
/
LCA2class.cpp
File metadata and controls
94 lines (87 loc) · 1.75 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include <bits/stdc++.h>
using namespace std;
class LCA
{
public:
LCA(int n) {
input(n);
dfs(0, -1);
buildTable();
}
int Lca(int x, int y) {
return lca(x, y) + 1;
}
private:
static const int MAXN = 100000 + 5;
vector<int> tree[2 * MAXN];
int depth[2 * MAXN];
int euler[4 * MAXN], eulerIdx = 0;
int rmq[20][4 * MAXN];
int first[2 * MAXN];
int logarithm[4 * MAXN + 1];
void dfs(int node, int parent) {
first[node] = eulerIdx;
euler[eulerIdx++] = node;
for (const int& son : tree[node]) {
if (son != parent) {
depth[son] = depth[node] + 1;
dfs(son, node);
euler[eulerIdx++] = node;
}
}
}
void buildTable() {
logarithm[1] = 0;
for (int i = 2; i <= eulerIdx; ++i) {
logarithm[i] = logarithm[i / 2] + 1;
}
for (int i = 0; i < eulerIdx; ++i) {
rmq[0][i] = euler[i];
}
for (int i = 1; (1 << i) <= eulerIdx; ++i) {
for (int j = 0; j + (1 << i) <= eulerIdx; ++j) {
if (depth[rmq[i - 1][j]] < depth[rmq[i - 1][j + (1 << (i - 1))]]) {
rmq[i][j] = rmq[i - 1][j];
}
else {
rmq[i][j] = rmq[i - 1][j + (1 << (i - 1))];
}
}
}
}
int lca(int a, int b) {
a = first[a]; b = first[b];
if (a > b) {
swap(a, b);
}
int y = logarithm[b - a + 1];
if (depth[rmq[y][a]] < depth[rmq[y][b - (1 << y) + 1]]) {
return rmq[y][a];
}
return rmq[y][b - (1 << y) + 1];
}
void input(int n) {
int x, y;
for (int i = 0; i < n - 1; ++i) {
scanf("%d%d", &x, &y);
x--; y--;
tree[x].push_back(y);
tree[y].push_back(x);
}
}
};
int main() {
int n, m, q;
scanf("%d", &n);
static LCA obj(n);
scanf("%d", &q);
while (q--) {
int a, b, c;
scanf("%d%d%d", &a,&b,&c);
a--; b--; c--;
int x = obj.Lca(a, b);
int y = obj.Lca(b, c);
int z = obj.Lca(a, c);
printf("%d\n", x^y^z);
}
}