data-structure/cumulative-sum.hpp

• View this file on GitHub
• Last update: 2023-07-08 14:12:19+09:00
• Include: #include "data-structure/cumulative-sum.hpp"

Depends on

• data-structure/base.hpp

Code

#pragma once

#include "./base.hpp"

#include <vector>

namespace matumoto {
  template <typename T>
  struct CumulativeSum {
    vector<T> data;
    CumulativeSum(int n): data(n + 1, 0) {}
    CumulativeSum(int n, T x): data(n + 1, x) {}

    void set(int k, T x) {
      data[k + 1] = x;
    }

    void add(int k, T x) {
      data[k + 1] = x;
    }

    T get(int k) {
      return data[k + 1];
    }

    void build() {
      for (int i = 0; i < static_cast<int>(data.size()) - 1; i++) {
        data[i + 1] += data[i];
      }
    }

    // [l,r)
    T query(int l, int r) {
      return data[r] - data[l];
    }
  };
} // namespace matumoto

#line 2 "data-structure/cumulative-sum.hpp"

#line 2 "data-structure/base.hpp"

namespace matumoto {
  using namespace std;
}
#line 4 "data-structure/cumulative-sum.hpp"

#include <vector>

namespace matumoto {
  template <typename T>
  struct CumulativeSum {
    vector<T> data;
    CumulativeSum(int n): data(n + 1, 0) {}
    CumulativeSum(int n, T x): data(n + 1, x) {}

    void set(int k, T x) {
      data[k + 1] = x;
    }

    void add(int k, T x) {
      data[k + 1] = x;
    }

    T get(int k) {
      return data[k + 1];
    }

    void build() {
      for (int i = 0; i < static_cast<int>(data.size()) - 1; i++) {
        data[i + 1] += data[i];
      }
    }

    // [l,r)
    T query(int l, int r) {
      return data[r] - data[l];
    }
  };
} // namespace matumoto

Back to top page