hkshakib

Untitled

Mar 22nd, 2020
200
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.38 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. #define mx 10000
  3. using namespace std;
  4. using ll = long long;
  5. ll arr[mx];
  6. struct info {
  7. ll prop, sum;
  8. } tree[mx * 4];
  9.  
  10. void init(int node, int b, int e)
  11. {
  12. if (b == e) {
  13. tree[node].sum = arr[b];
  14. return;
  15. }
  16. int Left = node * 2;
  17. int Right = node * 2 + 1;
  18. int mid = (b + e) / 2;
  19. init(Left, b, mid);
  20. init(Right, mid + 1, e);
  21. tree[node].sum = tree[Left].sum + tree[Right].sum;
  22. }
  23.  
  24. void update(int node, int b, int e, int i, int j, ll x)
  25. {
  26. if (i > e || j < b)
  27. return;
  28. if (b >= i && e <= j)
  29. {
  30. tree[node].sum += ((e - b + 1) * x);
  31. tree[node].prop += x;
  32. return;
  33. }
  34. int Left = node * 2;
  35. int Right = (node * 2) + 1;
  36. int mid = (b + e) / 2;
  37. update(Left, b, mid, i, j, x);
  38. update(Right, mid + 1, e, i, j, x);
  39. tree[node].sum = tree[Left].sum + tree[Right].sum + (e - b + 1) * tree[node].prop;
  40. }
  41.  
  42. ll query(int node, int b, int e, int i, int j, ll carry = 0)
  43. {
  44. if (i > e || j < b)
  45. return 0;
  46.  
  47. if (b >= i and e <= j)
  48. return tree[node].sum + carry * (e - b + 1);
  49.  
  50. int Left = node << 1;
  51. int Right = (node << 1) + 1;
  52. int mid = (b + e) >> 1;
  53.  
  54. ll p1 = query(Left, b, mid, i, j, carry + tree[node].prop);
  55. ll p2 = query(Right, mid + 1, e, i, j, carry + tree[node].prop);
  56.  
  57. return p1 + p2;
  58. }
Advertisement
Add Comment
Please, Sign In to add comment