累積和
(data_structure/sequence/cumulative_sum.hpp)

• View this file on GitHub
• Last update: 2022-07-17 06:33:45+09:00
• Include: #include "data_structure/sequence/cumulative_sum.hpp"

Verified with

• test/library_checker/data_structure/static_range_sum.test.cpp

Code

#pragma once
/**
 * @brief 累積和
*/
template <typename T>
struct cumulative_sum{
  vector<T> S;
  function<T(T, T)> f;
  T E;
  cumulative_sum(){
  }
  cumulative_sum(vector<T> A, function<T(T, T)> f, T E): f(f), E(E){
    int N = A.size();
    S = vector<T>(N + 1);
    S[0] = E;
    for (int i = 0; i < N; i++){
      S[i + 1] = f(S[i], A[i]);
    }
  }
  T get(int i){
    return S[i];
  }
};

#line 2 "data_structure/sequence/cumulative_sum.hpp"
/**
 * @brief 累積和
*/
template <typename T>
struct cumulative_sum{
  vector<T> S;
  function<T(T, T)> f;
  T E;
  cumulative_sum(){
  }
  cumulative_sum(vector<T> A, function<T(T, T)> f, T E): f(f), E(E){
    int N = A.size();
    S = vector<T>(N + 1);
    S[0] = E;
    for (int i = 0; i < N; i++){
      S[i + 1] = f(S[i], A[i]);
    }
  }
  T get(int i){
    return S[i];
  }
};

Back to top page