danielvitor23

Dynamic GCD

May 12th, 2023
614
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 4.24 KB | Source Code | 0 0
  1. #include <bits/stdc++.h>
  2.  
  3. using namespace std;
  4.  
  5. const int INF = 0x3f3f3f3f3f3f3f3fLL;
  6.  
  7. int n;
  8. vector<int> tr, tr_sub, a, v;
  9.  
  10. int timer;
  11. vector<int> sub, h, dad, pos;
  12.  
  13. vector<vector<int>> gr;
  14.  
  15. // ------- SegTree -----------
  16. int join(int a, int b) {
  17.   return gcd(a, b);
  18. }
  19.  
  20. void build_seg(int node, int l, int r) {
  21.   if (l == r) {
  22.     tr[node] = v[l];
  23.     return;
  24.   }
  25.   int lc = node << 1;
  26.   int mid = l + (r - l) / 2;
  27.   build_seg(lc, l, mid);
  28.   build_seg(lc+1, mid+1, r);
  29. }
  30.  
  31. int query(int node, int l, int r, int idx) {
  32.   if (l == r) return tr[node];
  33.   int lc = node << 1;
  34.   int mid = l + (r - l) / 2;
  35.   if (idx <= mid)
  36.     return tr[node] + query(lc, l, mid, idx);
  37.   else
  38.     return tr[node] + query(lc+1, mid+1, r, idx);
  39. }
  40.  
  41. void update(int node, int l, int r, int ul, int ur, int val) {
  42.   if (r < l or ur < ul or ur < l or r < ul) return;
  43.   if (ul <= l and r <= ur) {
  44.     tr[node] += val;
  45.     return;
  46.   }
  47.   int lc = node << 1;
  48.   int mid = l + (r - l) / 2;
  49.   update(lc, l, mid, ul, min(mid, ur), val);
  50.   update(lc+1, mid+1, r, max(mid+1, ul), ur, val);
  51. }
  52. // ------- SegTree -----------
  53.  
  54. // ------- SegTree -----------
  55. void build_sub(int node, int l, int r) {
  56.   if (l == r) {
  57.     tr_sub[node] = l == 1 ? v[l] : v[l] - v[l-1];
  58.     return;
  59.   }
  60.   int lc = node << 1;
  61.   int mid = l + (r - l) / 2;
  62.   build_sub(lc, l, mid);
  63.   build_sub(lc+1, mid+1, r);
  64.   tr_sub[node] = join(tr_sub[lc], tr_sub[lc+1]);
  65. }
  66.  
  67. void update_sub(int node, int l, int r, int idx, int val) {
  68.   if (l == r) {
  69.     tr_sub[node] += val;
  70.     return;
  71.   }
  72.   int lc = node << 1;
  73.   int mid = l + (r - l) / 2;
  74.   if (idx <= mid)
  75.     update_sub(lc, l, mid, idx, val);
  76.   else
  77.     update_sub(lc+1, mid+1, r, idx, val);
  78.   tr_sub[node] = join(tr_sub[lc], tr_sub[lc+1]);
  79. }
  80.  
  81. int query_sub(int node, int l, int r, int ql, int qr) {
  82.   // if (r < l or qr < ql or qr < l or r < ql) return 0;
  83.   if (ql <= l and r <= qr) return tr_sub[node];
  84.   int lc = node << 1;
  85.   int mid = l + (r - l) / 2;
  86.   if (qr <= mid) return query_sub(lc, l, mid, ql, min(mid, qr));
  87.   if (mid+1 <= ql) return query_sub(lc+1, mid+1, r, max(mid+1, ql), qr);
  88.   return join(
  89.     query_sub(lc, l, mid, ql, min(mid, qr)),
  90.     query_sub(lc+1, mid+1, r, max(mid+1, ql), qr)
  91.   );
  92. }
  93.  
  94. void update(int l, int r, int val) {
  95.   update(1, 1, n, l, r, val);
  96.   update_sub(1, 1, n, l, val);
  97.   update_sub(1, 1, n, r+1, -val);
  98. }
  99.  
  100. int query(int l, int r) {
  101.   if (l == r) {
  102.     return query(1, 1, n, l);
  103.   }
  104.   return join(
  105.     query_sub(1, 1, n, l + 1, r),
  106.     query(1, 1, n, l)
  107.   );
  108. }
  109. // ------- SegTree -----------
  110.  
  111. // ------- HLD -----------
  112. void dfs(int u, int par = -1) {
  113.   sub[u] = 1;
  114.   for (int &to : gr[u]) if (to != par) {
  115.     dad[to] = u;
  116.     dfs(to, u);
  117.     sub[u] += sub[to];
  118.     if (sub[to] > sub[gr[u][0]] or gr[u][0] == par)
  119.       swap(gr[u][0], to);
  120.   }
  121. }
  122.  
  123. void build_hld(int u, int par = -1) {
  124.   pos[u] = ++timer;
  125.   v[pos[u]] = a[u];
  126.   for (int to : gr[u]) if (to != par) {
  127.     h[to] = (to == gr[u][0] ? h[u] : to);
  128.     build_hld(to, u);
  129.   }
  130. }
  131.  
  132. int query_path(int a, int b) {
  133.   if (pos[a] < pos[b]) swap(a, b);
  134.  
  135.   if (h[a] == h[b]) return query(pos[b], pos[a]);
  136.  
  137.   return join(query(pos[h[a]], pos[a]), query_path(dad[h[a]], b));
  138. }
  139.  
  140. void update_path(int a, int b, int d) {
  141.   if (pos[a] < pos[b]) swap(a, b);
  142.  
  143.   if (h[a] == h[b]) return void(update(pos[b], pos[a], d));
  144.  
  145.   update(pos[h[a]], pos[a], d);
  146.   update_path(dad[h[a]], b, d);
  147. }
  148. // ------- HLD -----------
  149.  
  150. int main() {
  151.   cin.tie(0)->sync_with_stdio(0);
  152.  
  153.   cin >> n;
  154.  
  155.   gr.assign(n, vector<int>());
  156.   tr.assign(n << 2, 0);
  157.   tr_sub.assign(n << 2, 0);
  158.   a.assign(n, 0);
  159.   v.assign(n + 1, 0);
  160.  
  161.   sub.assign(n, 0);
  162.   h.assign(n, 0);
  163.   dad.assign(n, -1);
  164.   pos.assign(n, 0);
  165.  
  166.   for (int i = 0, a, b; i < n-1; ++i) {
  167.     cin >> a >> b;
  168.     gr[a].push_back(b);
  169.     gr[b].push_back(a);
  170.   }
  171.  
  172.   for (int i = 0; i < n; ++i) {
  173.     cin >> a[i];
  174.   }
  175.  
  176.   dfs(0);
  177.   build_hld(0);
  178.  
  179.   build_seg(1, 1, n);
  180.   build_sub(1, 1, n);
  181.  
  182.   int q; cin >> q;
  183.   while (q--) {
  184.     char c; cin >> c;
  185.     if (c == 'F') {
  186.       int a, b; cin >> a >> b;
  187.  
  188.       cout << abs(query_path(a, b)) << '\n';
  189.     } else {
  190.       int a, b, d; cin >> a >> b >> d;
  191.       if (!d) continue;
  192.  
  193.       update_path(a, b, d);
  194.     }
  195.   }
  196.  
  197. }
Advertisement
Add Comment
Please, Sign In to add comment