test/yosupo/convolution/lcm_convolution.test.cpp

• View this file on GitHub
• Last update: 2025-05-29 15:31:41+09:00
• Problem: https://judge.yosupo.jp/problem/lcm_convolution

Depends on

• Montgomery modular multiplication (lib/internal/internal_math.hpp)
• エラトステネスの篩 (lib/math/eratosthenes.hpp)
• lib/math/modint.hpp

Code

// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/lcm_convolution
#include <iostream>
#include <vector>
#include "math/eratosthenes.hpp"
#include "math/modint.hpp"

eratosthenes pn;
using Mint = modint998;

int main(void) {
    int n;
    std::cin >> n;
    std::vector<Mint> a(n + 1), b(n + 1);
    for (int i = 1; i <= n; ++i) std::cin >> a[i];
    for (int i = 1; i <= n; ++i) std::cin >> b[i];
    for (int p = 2; p <= n; ++p) {
        if (!pn.is_prime(p)) continue;
        for (int i = 1; i * p <= n; ++i) {
            a[i * p] += a[i];
            b[i * p] += b[i];
        }
    }
    for (int i = 1; i <= n; ++i) a[i] *= b[i];
    for (int p = 2; p <= n; ++p) {
        if (!pn.is_prime(p)) continue;
        for (int i = n / p; i >= 1; --i) a[i * p] -= a[i];
    }
    for (int i = 1; i <= n; ++i) std::cout << a[i] << (i == n ? '\n' : ' ');

    return 0;
}

Traceback (most recent call last):
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/resolver.py", line 291, in resolve
    bundled_code = language.bundle(path, basedir=basedir)
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/verify/languages/cplusplus.py", line 242, in bundle
    bundler.update(path)
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/verify/languages/cplusplus_bundle.py", line 479, in update
    self._resolve(pathlib.Path(included), included_from=path)
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/verify/languages/cplusplus_bundle.py", line 286, in _resolve
    raise BundleErrorAt(path, -1, "no such header")
competitive_verifier.oj.verify.languages.cplusplus_bundle.BundleErrorAt: math/eratosthenes.hpp: line -1: no such header

Test cases

Env	Name	Status	Elapsed	Memory
g++	all_zero_00	AC	4 ms	4 MB
g++	example_00	AC	4 ms	4 MB
g++	max_random_00	AC	631 ms	11 MB
g++	max_random_01	AC	649 ms	11 MB
g++	near_prime_00	AC	633 ms	11 MB
g++	near_prime_01	AC	646 ms	11 MB
g++	near_prime_02	AC	634 ms	11 MB
g++	near_prime_squared_00	AC	653 ms	11 MB
g++	near_prime_squared_01	AC	642 ms	11 MB
g++	near_prime_squared_02	AC	629 ms	11 MB
g++	random_00	AC	250 ms	6 MB
g++	random_01	AC	294 ms	7 MB
g++	random_02	AC	369 ms	8 MB
g++	small_00	AC	4 ms	4 MB
g++	small_01	AC	4 ms	4 MB
g++	small_02	AC	4 ms	4 MB
g++	small_03	AC	4 ms	4 MB
g++	small_04	AC	4 ms	4 MB
g++	small_05	AC	4 ms	4 MB
g++	small_06	AC	4 ms	4 MB
g++	small_07	AC	4 ms	4 MB
g++	small_08	AC	4 ms	4 MB
g++	small_09	AC	4 ms	4 MB
g++	small_10	AC	4 ms	4 MB
g++	small_11	AC	4 ms	4 MB
g++	small_12	AC	4 ms	4 MB
g++	small_13	AC	4 ms	4 MB
g++	small_14	AC	4 ms	4 MB
g++	small_15	AC	4 ms	4 MB

Back to top page