lalalalalalalaalalla

Untitled

Dec 18th, 2019
103
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 3.32 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3. #include <string>
  4. #include <iomanip>
  5. #include <queue>
  6. #include <cmath>
  7. #include <algorithm>
  8. #include <unordered_map>
  9. #include <tuple>
  10. #include <iomanip>
  11. #include <stdio.h>
  12. #include <map>
  13. #include <bitset>
  14. #include <set>
  15. #include <stack>
  16. #include <queue>
  17. #include <unordered_set>
  18. #include <cassert>
  19. #include <stdlib.h>
  20. #include <time.h>
  21. #include <random>
  22.  
  23.  
  24. //#pragma GCC optimize("Ofast,no-stack-protector")
  25. //#pragma GCC target("sse,sse2,sse3,sse3,sse4")
  26. //#pragma GCC optimize("unroll-loops")
  27. //#pragma GCC optimize("fast-math")
  28. //#pragma GCC target("avx2")
  29. //#pragma GCC optimize("section-anchors")
  30. //#pragma GCC optimize("profile-values,profile-reorder-functions,tracer")
  31. //#pragma GCC optimize("vpt")
  32. //#pragma GCC optimize("rename-registers")
  33. //#pragma GCC optimize("move-loop-invariants")
  34. //#pragma GCC optimize("unswitch-loops")
  35. //#pragma GCC optimize("function-sections")
  36. //#pragma GCC optimize("data-sections")
  37.  
  38. //#define int long long
  39. #define ll long long
  40. #define ull unsigned long long
  41. #define all(a) (a).begin(), (a).end()
  42. #define pii pair<int, int>
  43. #define pb push_back
  44. #define ld long double
  45.  
  46. using namespace std;
  47.  
  48. //const int INF = 1e17;
  49. //const int mod = 2600000069;
  50. //const int p = 179;
  51. const int MAXN = 2e5 + 1;
  52. const int logn = 19;
  53.  
  54. int up[MAXN][logn], f[MAXN], tin[MAXN], tout[MAXN], par[MAXN], nxt[MAXN];
  55. vector<int> g[MAXN];
  56. int ind = 1, timer = 0;
  57.  
  58. struct event {
  59. int t, u, v;
  60. event(int u_) {
  61. t = 0;
  62. u = u_;
  63. }
  64. event(int u_, int v_) {
  65. t = 1;
  66. u = u_;
  67. v = v_;
  68. }
  69. };
  70.  
  71. vector<event> c;
  72.  
  73. void dfs(int v, int p) {
  74. tin[v] = timer++;
  75. up[v][0] = p;
  76. par[v] = p;
  77. nxt[v] = v;
  78. for (int i = 1; i < logn; i++) up[v][i] = up[up[v][i - 1]][i - 1];
  79. for (auto k : g[v]) {
  80. if (k != p) dfs(k, v);
  81. }
  82. tout[v] = timer;
  83. }
  84.  
  85. bool ancestor(int v, int u) { // v -> u
  86. return tin[v] <= tin[u] && tout[v] >= tout[u];
  87. }
  88.  
  89. int get_par(int v) {
  90. if (nxt[v] == v) return v;
  91. return nxt[v] = get_par(nxt[v]);
  92. }
  93.  
  94. int lca(int v, int u) {
  95. if (ancestor(u, v)) return u;
  96. if (ancestor(v, u)) return v;
  97. for (int i = logn - 1; i >= 0; i--) {
  98. if (!ancestor(up[v][i], u)) {
  99. v = up[v][i];
  100. }
  101. }
  102. v = up[v][0];
  103. // cout <<"v: " << v << "\n";
  104. if (f[v]) return get_par(v);
  105. return v;
  106. }
  107.  
  108. signed main() {
  109. ios_base::sync_with_stdio(0);
  110. cin.tie(0);
  111. cout.tie(0);
  112. for (int i = 0; i < logn; i++) {
  113. up[0][i] = 0;
  114. }
  115. nxt[0] = 0;
  116. int n;
  117. cin >> n;
  118. int u, v;
  119. char t;
  120. for (int i = 0; i < n; i++) {
  121. cin >> t;
  122. if (t == '+') {
  123. cin >> v;
  124. v--;
  125. g[ind].pb(v);
  126. g[v].pb(ind);
  127. ind++;
  128. } else if (t == '-') {
  129. cin >> v;
  130. v--;
  131. c.pb(event(v));
  132. } else {
  133. cin >> v >> u;
  134. v--;u--;
  135. c.pb(event(u, v));
  136. }
  137. }
  138. dfs(0, 0);
  139. for (auto k : c) {
  140. if (k.t) {
  141. cout << lca(k.u, k.v) + 1 << "\n";
  142. } else {
  143. f[k.u] = 1;
  144. nxt[k.u] = par[k.u];
  145. }
  146. }
  147. }
  148. /*
  149. 11
  150. + 1
  151. + 1
  152. + 2
  153. ? 2 3
  154. ? 1 3
  155. ? 2 4
  156. + 4
  157. + 4
  158. - 4
  159. ? 5 6
  160. ? 5 5
  161. -> 1 1 2 2 5
  162. */
Add Comment
Please, Sign In to add comment