4eyes4u

Segment tree: lazy propagation

Apr 6th, 2016
157
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.74 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. #define mid (left+right)/2
  3. #define lson 2*idx,left,mid
  4. #define rson 2*idx+1,mid+1,right
  5. using namespace std;
  6.  
  7. const int N=4e5+10;
  8.  
  9. long long st[N], lazy[N];
  10. int t;
  11.  
  12. void init (int idx, int left, int right)
  13. {
  14.     st[idx]=lazy[idx]=0;
  15.     if (left<right) init(lson), init(rson);
  16. }
  17.  
  18. void propagate (int idx, int left, int right)
  19. {
  20.     if (lazy[idx]!=0)
  21.     {
  22.         st[idx]+=lazy[idx]*(right-left+1);
  23.         if (left<right)
  24.         {
  25.             lazy[2*idx]+=lazy[idx];
  26.             lazy[2*idx+1]+=lazy[idx];
  27.         }
  28.         lazy[idx]=0;
  29.     }
  30. }
  31.  
  32. void update (int idx, int left, int right, int l, int r, long long val)
  33. {
  34.     propagate(idx, left, right);
  35.     if (left>right || left>r || right<l) return;
  36.  
  37.     if (l<=left && right<=r)
  38.     {
  39.         st[idx]+=val*(right-left+1);
  40.         if (left<right)
  41.         {
  42.             lazy[2*idx]+=val;
  43.             lazy[2*idx+1]+=val;
  44.         }
  45.         return;
  46.     }
  47.     update(lson, l, r, val);
  48.     update(rson, l, r, val);
  49.     st[idx]=st[2*idx]+st[2*idx+1];
  50. }
  51.  
  52. long long query (int idx, int left, int right, int l, int r)
  53. {
  54.     propagate(idx, left, right);
  55.     if (left>right || left>r || right<l) return 0;
  56.  
  57.     if (l<=left && right<=r) return st[idx];
  58.     return (query(lson, l, r)+query(rson, l, r));
  59. }
  60.  
  61. void solve ()
  62. {
  63.     int n, q, op, l, r;
  64.     long long val;
  65.  
  66.     scanf ("%d%d", &n, &q);
  67.     init(1, 1, n);
  68.  
  69.     while (q--)
  70.     {
  71.         scanf ("%d%d%d", &op, &l, &r);
  72.         if (op==1) printf ("%lld\n", query(1, 1, n, l, r));
  73.         else
  74.         {
  75.             scanf ("%lld", &val);
  76.             update (1, 1, n, l, r, val);
  77.         }
  78.     }
  79. }
  80.  
  81. int main()
  82. {
  83.     scanf ("%d", &t);
  84.     while (t--) solve();
  85.  
  86.     return 0;
  87. }
Advertisement
Add Comment
Please, Sign In to add comment