formalpowerseries/base.cpp

• View this file on GitHub
• Last update: 2021-03-25 09:21:12+09:00
• Link: http://beet-aizu.hatenablog.com/entry/2019/09/27/224701

Depends on

• combinatorics/enumeration.cpp

Required by

• combinatorics/bernoulli.cpp
• combinatorics/partition.cpp
• combinatorics/stirling1st.cpp
• combinatorics/stirling2nd.cpp
• formalpowerseries/998244353.cpp
• formalpowerseries/diff.cpp
• formalpowerseries/div.cpp
• formalpowerseries/exp.cpp
• formalpowerseries/integral.cpp
• formalpowerseries/inv.cpp
• formalpowerseries/log.cpp
• formalpowerseries/mod.cpp
• formalpowerseries/pow.cpp
• formalpowerseries/shift.cpp
• formalpowerseries/sqrt.cpp
• includes/formalpowerseries.h
• polynomial/interpolate.cpp
• polynomial/multieval.cpp

Verified with

• test/aoj/2985.garner.test.cpp
• test/aoj/2985.test.cpp
• test/aoj/3072.test.cpp
• test/aoj/3084.test.cpp
• test/yosupo/bernoulli_number.test.cpp
• test/yosupo/exp_of_formal_power_series.test.cpp
• test/yosupo/inv_of_formal_power_series.test.cpp
• test/yosupo/log_of_formal_power_series.test.cpp
• test/yosupo/multipoint_evaluation.test.cpp
• test/yosupo/partition_function.test.cpp
• test/yosupo/polynomial_interpolation.test.cpp
• test/yosupo/polynomial_taylor_shift.test.cpp
• test/yosupo/pow_of_formal_power_series.test.cpp
• test/yosupo/sharp_p_subset_sum.test.cpp
• test/yosupo/sqrt_of_formal_power_series.test.cpp
• test/yosupo/stirling_number_of_the_first_kind.test.cpp
• test/yosupo/stirling_number_of_the_second_kind.test.cpp
• test/yukicoder/2744.test.cpp
• test/yukicoder/5633.test.cpp

Code

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

#define call_from_test
#include "../combinatorics/enumeration.cpp"
#undef call_from_test

#endif
// http://beet-aizu.hatenablog.com/entry/2019/09/27/224701
//BEGIN CUT HERE
template<typename M_>
struct FormalPowerSeries : Enumeration<M_> {
  using M = M_;
  using super = Enumeration<M>;
  using super::fact;
  using super::finv;
  using super::invs;

  using Poly = vector<M>;
  using Conv = function<Poly(Poly, Poly)>;
  Conv conv;
  FormalPowerSeries(Conv conv):conv(conv){}

  Poly pre(const Poly &as,int deg){
    return Poly(as.begin(),as.begin()+min((int)as.size(),deg));
  }

  Poly add(Poly as,Poly bs){
    int sz=max(as.size(),bs.size());
    Poly cs(sz,M(0));
    for(int i=0;i<(int)as.size();i++) cs[i]+=as[i];
    for(int i=0;i<(int)bs.size();i++) cs[i]+=bs[i];
    return cs;
  }

  Poly sub(Poly as,Poly bs){
    int sz=max(as.size(),bs.size());
    Poly cs(sz,M(0));
    for(int i=0;i<(int)as.size();i++) cs[i]+=as[i];
    for(int i=0;i<(int)bs.size();i++) cs[i]-=bs[i];
    return cs;
  }

  Poly mul(Poly as,Poly bs){
    return conv(as,bs);
  }

  Poly mul(Poly as,M k){
    for(auto &a:as) a*=k;
    return as;
  }

  bool is_zero(Poly as){
    return as==Poly(as.size(),0);
  }

  void shrink(Poly &as){
    assert(not is_zero(as));
    while(as.back()==M(0)) as.pop_back();
  }

  // F(0) must not be 0
  Poly inv(Poly as,int deg);

  // not zero
  Poly div(Poly as,Poly bs);

  // not zero
  Poly mod(Poly as,Poly bs);

  // F(0) must be 1
  Poly sqrt(Poly as,int deg);

  Poly diff(Poly as);
  Poly integral(Poly as);

  // F(0) must be 1
  Poly log(Poly as,int deg);

  // F(0) must be 0
  Poly exp(Poly as,int deg);

  // not zero
  Poly pow(Poly as,long long k,int deg);

  // x <- x + c
  Poly shift(Poly as,M c);
};
//END CUT HERE
#ifndef call_from_test
//INSERT ABOVE HERE
signed main(){
  return 0;
}
#endif

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.11.3/x64/lib/python3.11/site-packages/onlinejudge_verify/documentation/build.py", line 71, in _render_source_code_stat
    bundled_code = language.bundle(stat.path, basedir=basedir, options={'include_paths': [basedir]}).decode()
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.11.3/x64/lib/python3.11/site-packages/onlinejudge_verify/languages/cplusplus.py", line 187, in bundle
    bundler.update(path)
  File "/opt/hostedtoolcache/Python/3.11.3/x64/lib/python3.11/site-packages/onlinejudge_verify/languages/cplusplus_bundle.py", line 400, in update
    raise BundleErrorAt(path, i + 1, "unable to process #include in #if / #ifdef / #ifndef other than include guards")
onlinejudge_verify.languages.cplusplus_bundle.BundleErrorAt: formalpowerseries/base.cpp: line 6: unable to process #include in #if / #ifdef / #ifndef other than include guards

Back to top page