test/yosupo/sum_of_multiplicative_function.test.cpp

View this file on GitHub
Last update: 2024-09-28 01:48:34+09:00
Problem: https://judge.yosupo.jp/problem/sum_of_multiplicative_function

Depends on

Sum of Multiplicative Function (src/math/sum_of_multiplicative_function.hpp)

Code

#define PROBLEM "https://judge.yosupo.jp/problem/sum_of_multiplicative_function"

#include "math/sum_of_multiplicative_function.hpp"
#include <iostream>
#include "atcoder/modint"

using mint = atcoder::static_modint<469762049>;

long long a, b;

mint f(long long p, long long e) { return mint(a) * e + mint(b) * p; }

void solve() {
    long long N;
    std::cin >> N >> a >> b;

    sum_of_multiplicative_function<mint, f> smf(N);
    auto S0 = smf.counting_primes();
    auto S1 = smf.summing_primes();
    for (int i = 0; i < int(S1.size()); i++) S1[i] = S1[i] * b + S0[i] * a;
    auto ans = smf.solve(S1);

    std::cout << ans.val() << "\n";
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int T;
    std::cin >> T;
    for (; T--;) solve();
    return 0;
}

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.13.3/x64/lib/python3.13/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.13.3/x64/lib/python3.13/site-packages/onlinejudge_verify/languages/cplusplus.py", line 187, in bundle
    bundler.update(path)
    ~~~~~~~~~~~~~~^^^^^^
  File "/opt/hostedtoolcache/Python/3.13.3/x64/lib/python3.13/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.13.3/x64/lib/python3.13/site-packages/onlinejudge_verify/languages/cplusplus_bundle.py", line 260, in _resolve
    raise BundleErrorAt(path, -1, "no such header")
onlinejudge_verify.languages.cplusplus_bundle.BundleErrorAt: math/sum_of_multiplicative_function.hpp: line -1: no such header