lalalalalalalaalalla

Untitled

Dec 18th, 2019
96
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 3.71 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 = 1e5 + 1;
  52. const int logn = 18;
  53.  
  54. int tin[MAXN], tout[MAXN], up[MAXN][logn];
  55. vector<int> g[MAXN];
  56. int timer = 0;
  57.  
  58. void dfs(int v, int p) {
  59. up[v][0] = p;
  60. tin[v] = timer++;
  61. for (int i = 1; i < logn; i++) {
  62. up[v][i] = up[up[v][i - 1]][i - 1];
  63. }
  64. for (auto u : g[v]) {
  65. if (u != p) dfs(u, v);
  66. }
  67. tout[v] = timer++;
  68. }
  69.  
  70. bool ancestor(int v, int u) {
  71. return tin[v] <= tin[u] && tout[v] >= tout[u];
  72. }
  73.  
  74. int lca(int v, int u) {
  75. if (ancestor(u, v)) return u;
  76. if (ancestor(v, u)) return v;
  77. for (int i = logn - 1; i >= 0; i--) {
  78. if (!ancestor(up[v][i], u)) v = up[v][i];
  79. }
  80. return up[v][0];
  81. }
  82.  
  83. int cur = 0;
  84.  
  85. int ans() {
  86. int u, v;
  87. cin >> u >> v;
  88. u--;v--;
  89. if (u == v) return u;
  90. int uv = lca(u, v), ur = lca(u, cur), vr = lca(v, cur);
  91. // cout << "vr : " << vr << "\n";
  92. // cout << "ur : " << ur << "\n";
  93. int help = 0;
  94. if (!ancestor(uv, cur) || ancestor(cur, uv)) return uv;
  95. if (ancestor(u, v)) {
  96. if (ancestor(vr, v) && ancestor(u, vr)) return vr;
  97. if (ancestor(cur, u)) return u;
  98. if (ancestor(v, cur)) return v;
  99. while(1) help++;
  100. }
  101. if (ancestor(v, u)) {
  102. if (ancestor(ur, u) && ancestor(v, ur)) return ur;
  103. if (ancestor(cur, v)) return v;
  104. if (ancestor(u, cur)) return u;
  105. while(1) help++;
  106. }
  107. if (ancestor(u, cur)) return u;
  108. if (ancestor(v, cur)) return v;
  109. if (ancestor(uv, ur) && ur != uv) return ur;
  110. if (ancestor(uv, vr) && uv != vr) return vr;
  111. if (ancestor(uv, ur)) return ur;
  112. // if (ancestor(uv, vr)) return vr;
  113. return 123456789;
  114. }
  115.  
  116. signed main() {
  117. ios_base::sync_with_stdio(0);
  118. cin.tie(0);
  119. cout.tie(0);
  120. int n;
  121. cin >> n;
  122. while (n != 0) {
  123. int u, v;
  124. cur = 0;
  125. for (int i = 0; i < n; i++) g[i].clear();
  126. for (int i = 0; i < n - 1; i++) {
  127. cin >> u >> v;
  128. u--;v--;
  129. g[u].pb(v);
  130. g[v].pb(u);
  131. }
  132. dfs(0, 0);
  133. int q;
  134. cin >> q;
  135. char t;
  136. while (q--) {
  137. cin >> t;
  138. if (t == '?') cout << ans() + 1 << "\n";
  139. else {
  140. cin >> cur;
  141. cur--;
  142. }
  143. }
  144. cin >> n;
  145. }
  146. }
  147. /*
  148. 9
  149. 1 2
  150. 1 3
  151. 2 4
  152. 2 5
  153. 3 6
  154. 3 7
  155. 6 8
  156. 6 9
  157. 10
  158. ? 4 5
  159. ? 5 6
  160. ? 8 7
  161. ! 6
  162. ? 8 7
  163. ? 4 5
  164. ? 4 7
  165. ? 5 9
  166. ! 2
  167. ? 4 3
  168. 0
  169. */
Add Comment
Please, Sign In to add comment