test/yosupo/sharp_p_subset_sum.test.cpp

• View this file on GitHub
• Last update: 2021-03-25 09:46:10+09:00
• Problem: https://judge.yosupo.jp/problem/sharp_p_subset_sum

Depends on

• combinatorics/enumeration.cpp
• convolution/numbertheoretictransform.cpp
• formalpowerseries/base.cpp
• formalpowerseries/diff.cpp
• formalpowerseries/exp.cpp
• formalpowerseries/integral.cpp
• formalpowerseries/inv.cpp
• formalpowerseries/log.cpp
• mod/mint.cpp

Code

// verification-helper: PROBLEM https://judge.yosupo.jp/problem/sharp_p_subset_sum

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

#define call_from_test
#include "../../mod/mint.cpp"
#include "../../convolution/numbertheoretictransform.cpp"
#include "../../combinatorics/enumeration.cpp"
#include "../../formalpowerseries/base.cpp"
#include "../../formalpowerseries/inv.cpp"
#include "../../formalpowerseries/integral.cpp"
#include "../../formalpowerseries/diff.cpp"
#include "../../formalpowerseries/log.cpp"
#include "../../formalpowerseries/exp.cpp"
#undef call_from_test

signed main(){
  cin.tie(0);
  ios::sync_with_stdio(0);

  int n,t;
  cin>>n>>t;

  NTT<2> ntt;
  using M = NTT<2>::M;
  auto conv=[&](auto as,auto bs){return ntt.multiply(as,bs);};
  FormalPowerSeries<M> FPS(conv);
  decltype(FPS)::init(t+1);

  vector<int> cnt(t+1,0);
  for(int i=0;i<n;i++){
    int s;
    cin>>s;
    cnt[s]++;
  }

  vector<M> as(t+1);
  for(int s=1;s<=t;s++){
    for(int k=1;s*k<=t;k++){
      if(k&1) as[s*k]+=M(cnt[s])*decltype(FPS)::Invs(k);
      else as[s*k]-=M(cnt[s])*decltype(FPS)::Invs(k);
    }
  }

  auto bs=FPS.exp(as,t+1);
  for(int i=1;i<=t;i++){
    if(i>1) cout<<" ";
    cout<<bs[i];
  }
  cout<<endl;
  return 0;
}

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 401, in update
    self.update(self._resolve(pathlib.Path(included), included_from=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: convolution/numbertheoretictransform.cpp: line 6: unable to process #include in #if / #ifdef / #ifndef other than include guards

Back to top page