src/dp/cumulative_sum.hpp

View this file on GitHub
Last update: 2022-12-23 01:30:30+09:00
Include: #include "src/dp/cumulative_sum.hpp"

Verified with

test/aoj/0516.test.cpp

Code

#pragma once

#include <bits/stdc++.h>
using namespace std;

template <typename T> struct CumulativeSum {
private:
  vector<T> s;

public:
  CumulativeSum(const vector<T> &nums) {
    int n = nums.size();
    s.resize(n + 1, 0);
    for (int i = 0; i < n; ++i) {
      s[i + 1] = s[i] + nums[i];
    }
  }

  // [l, r), l and r is 0-indexed.
  T sum(int l, int r) { return s[r] - s[l]; }
};

#line 2 "src/dp/cumulative_sum.hpp"

#include <bits/stdc++.h>
using namespace std;

template <typename T> struct CumulativeSum {
private:
  vector<T> s;

public:
  CumulativeSum(const vector<T> &nums) {
    int n = nums.size();
    s.resize(n + 1, 0);
    for (int i = 0; i < n; ++i) {
      s[i + 1] = s[i] + nums[i];
    }
  }

  // [l, r), l and r is 0-indexed.
  T sum(int l, int r) { return s[r] - s[l]; }
};