Advertisement
lalalalalalalaalalla

Untitled

Dec 18th, 2019
133
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 3.06 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;
  53. const int logn = 18;
  54.  
  55. int n;
  56. int tin[MAXN], tout[MAXN], up[MAXN][logn], val[MAXN], sum[MAXN];
  57. vector<int> g[MAXN];
  58. int timer = 0;
  59.  
  60. void dfs(int v, int p) {
  61. up[v][0] = p;
  62. for (int i = 1; i < logn; i++) {
  63. up[v][i] = up[up[v][i - 1]][i - 1];
  64. }
  65. tin[v] = timer++;
  66. sum[tin[v]] = sum[tin[p]] + val[v];
  67. for (auto u : g[v]) {
  68. if (u != p) dfs(u, v);
  69. }
  70. tout[v] = timer;
  71. }
  72.  
  73. bool ancestor(int u, int v) {
  74. return tin[u] <= tin[v] && tout[u] >= tout[v];
  75. }
  76.  
  77. int lca(int u, int v) {
  78. if (ancestor(u, v)) return u;
  79. if (ancestor(v, u)) return v;
  80. for (int i = logn - 1; i >= 0; i--) {
  81. if (!ancestor(up[v][i], u)) {
  82. v = up[v][i];
  83. }
  84. }
  85. return up[v][0];
  86. }
  87.  
  88. void ask() {
  89. int v, u;
  90. cin >> v >> u;
  91. u--;v--;
  92. int l = lca(u, v);
  93. if (l == u) {
  94. cout << sum[tin[v]] - (u != up[u][0] ? sum[tin[up[u][0]]] : 0) << "\n";
  95. } else if (l == v) {
  96. cout << sum[tin[u]] - (v != up[v][0] ? sum[tin[up[v][0]]] : 0) << "\n";
  97. } else {
  98. cout << sum[tin[u]] + sum[tin[v]] - (l != up[l][0] ? 2 * sum[tin[up[l][0]]] : 0) - val[l] << "\n";
  99. }
  100. }
  101.  
  102. void setv() {
  103. int v, x;
  104. cin >> v >> x;
  105. v--;
  106. for (int i = tin[v]; i < tout[v]; i++) {
  107. sum[i] += x - val[v];
  108. }
  109. val[v] = x;
  110. }
  111.  
  112. signed main() {
  113. ios_base::sync_with_stdio(0);
  114. cin.tie(0);
  115. cout.tie(0);
  116. cin >> n;
  117. for (int i = 0; i < n; i++) cin >> val[i];
  118. int u, v;
  119. for (int i = 0; i < n - 1; i++) {
  120. cin >> u >> v;
  121. u--;v--;
  122. g[u].pb(v);
  123. g[v].pb(u);
  124. }
  125. dfs(0, 0);
  126. int q;
  127. cin >> q;
  128. char t;
  129. while(q--) {
  130. cin >> t;
  131. if (t == '?') ask();
  132. else setv();
  133. }
  134. }
  135. /*
  136. */
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement