Advertisement
TrickmanOff

Fenwick with range updates

Feb 28th, 2020
273
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.03 KB | None | 0 0
  1. const int MAX_N = 1e5;
  2. int n;
  3. ll Mul[MAX_N], Add[MAX_N];
  4.  
  5. void suf_upd(int pos, ll mul, ll add) {
  6.     while(pos < n) {
  7.         Mul[pos] += mul;
  8.         Add[pos] += add;
  9.         pos |= (pos + 1);
  10.     }
  11. }
  12.  
  13. void upd(int l, int r, ll add) {
  14.     suf_upd(l, add, -add*(l-1));
  15.     suf_upd(r+1, -add, add*r);
  16. }
  17.  
  18. ll get(int pos) {
  19.     ll mul = 0, add = 0;
  20.     for(int i = pos; i >= 0; i = (i & (i+1)) - 1) {
  21.         mul += Mul[i];
  22.         add += Add[i];
  23.     }
  24.    
  25.     return mul * pos + add;
  26. }
  27.  
  28. ll get(int l, int r) {
  29.     return get(r) - (l > 0 ? get(l-1) : 0);
  30. }
  31.  
  32. int main() {
  33.     cin >> n;
  34.     for(int i = 0; i < n; i++) {
  35.         int a;
  36.         cin >> a;
  37.         upd(i, i, a);
  38.     }
  39.     int q;
  40.     cin >> q;
  41.     while(q--) {
  42.         char c;
  43.         cin >> c;
  44.         if(c == 'g') {
  45.             int pos;
  46.             cin >> pos;
  47.             cout << get(pos-1, pos-1) << ' ';
  48.         }
  49.         else {
  50.             int l, r, x;
  51.             cin >> l >> r >> x;
  52.             upd(l-1, r-1, x);
  53.         }
  54.     }
  55. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement