lalalalalalalaalalla

Untitled

Dec 18th, 2019
89
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 3.35 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.  
  47. using namespace std;
  48.  
  49. //const int INF = 1e17;
  50. //const int mod = 2600000069;
  51. //const int p = 179;
  52. const int MAXN = 1e5 + 10;
  53. const int logn = 23;
  54.  
  55. int up[MAXN][logn], sum[MAXN][logn], cost[MAXN], tin[MAXN], tout[MAXN];
  56. vector<int> g[MAXN];
  57. int timer = 0;
  58.  
  59. // (sum]
  60.  
  61. void dfs(int v, int p) {
  62. if (p != v) up[v][0] = p;
  63. sum[v][0] = cost[p];
  64. for (int i = 1; i < logn; i++) {
  65. up[v][i] = up[up[v][i - 1]][i - 1];
  66. sum[v][i] = sum[v][i - 1] + sum[up[v][i - 1]][i - 1];
  67. }
  68. tin[v] = timer++;
  69. for (auto u : g[v]) {
  70. if (u != p) {
  71. dfs(u, v);
  72. }
  73. }
  74. tout[v] = timer;
  75. }
  76.  
  77. bool ancestor(int u, int v) {
  78. return tin[u] <= tin[v] && tout[u] >= tout[v];
  79. }
  80.  
  81. void ask() {
  82. int u, v;
  83. cin >> u >> v;
  84. u--;v--;
  85. if (u == v) {
  86. cout << cost[u] << "\n";
  87. return;
  88. }
  89. int ans = cost[u] + cost[v];
  90. if (ancestor(u, v) || ancestor(v, u)) {
  91. if (ancestor(v, u)) swap(u, v);
  92. for (int i = logn - 1; i >= 0; i--) {
  93. if (!ancestor(up[v][i], u)) {
  94. ans += sum[v][i];
  95. v = up[v][i];
  96. }
  97. }
  98. } else {
  99. for (int i = logn - 1; i >= 0; i--) {
  100. if (!ancestor(up[v][i], u)) {
  101. ans += sum[v][i];
  102. v = up[v][i];
  103. }
  104. }
  105. ans += sum[v][0];
  106. v = up[v][0];
  107. for (int i = logn - 1; i >= 0; i--) {
  108. if (!ancestor(up[u][i], v)) {
  109. ans += sum[u][i];
  110. u = up[u][i];
  111. }
  112. }
  113. }
  114. cout << ans << "\n";
  115. }
  116.  
  117. signed main() {
  118. ios_base::sync_with_stdio(0);
  119. cin.tie(0);
  120. cout.tie(0);
  121. int n;
  122. cin >> n;
  123. for (int i = 0; i < n; i++) cin >> cost[i];
  124. int a, b;
  125. for (int i = 1; i < n; i++) {
  126. cin >> a >> b;
  127. a--;b--;
  128. g[a].pb(b);
  129. g[b].pb(a);
  130. }
  131. dfs(0, 0);
  132. // for (int i = 0; i < n; i++) cout << tin[i] << " ";
  133. // cout << "\n";
  134. // for (int i = 0; i < n; i++) cout << tout[i] << " ";
  135. // cout << "\n";
  136. int q;
  137. cin >> q;
  138. while (q--) ask();
  139. }
  140. /*
  141. 5
  142. 1 10 100 1000 10000
  143. 1 2
  144. 2 3
  145. 3 4
  146. 3 5
  147. 10
  148. 1 2
  149. */
Add Comment
Please, Sign In to add comment