verify/yosupo_math/prefix_sum_of_multiplicative_function.test.cpp

• View this file on GitHub
• Last update: 2025-06-04 11:53:52+09:00
• Problem: https://judge.yosupo.jp/problem/sum_of_multiplicative_function

Depends on

• math/enumerate_quotients.hpp
• math/frac_floor.hpp
• math/multiplicative_function/prefix_sum.hpp
• math/prime_table.hpp
• math/sqrt_floor.hpp
• modint/mont.hpp
• template/constant.hpp
• template/fastio.hpp
• template/io_util.hpp
• template/macros.hpp
• template/template.hpp
• template/type_alias.hpp
• type_traits/integral.hpp
• type_traits/io.hpp

Code

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

#include "../../math/multiplicative_function/prefix_sum.hpp"
#include "../../modint/mont.hpp"
#include "../../template/template.hpp"
using namespace std;

using mint = kk2::Mont<469762049>;

int main() {
    int t;
    kin >> t;
    rep (t) {
        i64 n;
        mint a, b;
        kin >> n >> a >> b;
        auto f = [&](i64 p, i64 e) -> mint { return a * e + b * p; };
        auto g1 = [&](i64) -> mint { return 1; };
        auto g2 = [&](i64 p) -> mint { return p; };
        kk2::PrefixSumOfMultiplicativeFunction<mint> ps(n);
        vc<mint> init1(ps.size()), init2(ps.size());
        const mint inv2 = mint(2).inv();
        rep (i, ps.size()) {
            init1[i] = ps.eq[i] - 1;
            init2[i] = mint(ps.eq[i]) * (ps.eq[i] + 1) * inv2 - 1;
        }
        ps.LucyDP(g1, init1);
        ps.LucyDP(g2, init2);
        rep (i, ps.size()) ps.prefix_sum_only_prime[i] = a * init1[i] + b * init2[i];
        ps.Min_25Sieve(f);
        kout << ps.prefix_sum.back() << "\n";
    }

    return 0;
}

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/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.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/cplusplus.py", line 187, in bundle
    bundler.update(path)
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/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.12.0/x64/lib/python3.12/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.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/cplusplus_bundle.py", line 312, in update
    raise BundleErrorAt(path, i + 1, "#pragma once found in a non-first line")
onlinejudge_verify.languages.cplusplus_bundle.BundleErrorAt: type_traits/integral.hpp: line 4: #pragma once found in a non-first line

Back to top page