Multidimensional Convolution (Truncated) (verify/poly/multivar.test.cpp)

• View this file on GitHub
• Last update: 2026-04-22 09:17:08-07:00
• Problem: https://judge.yosupo.jp/problem/multivariate_convolution

Depends on

• cp-algo/math/common.hpp
• cp-algo/math/cvector.hpp
• cp-algo/math/fft.hpp
• cp-algo/math/multivar.hpp
• cp-algo/number_theory/modint.hpp
• cp-algo/random/rng.hpp
• cp-algo/util/big_alloc.hpp
• cp-algo/util/checkpoint.hpp
• cp-algo/util/complex.hpp
• cp-algo/util/simd.hpp

Code

// @brief Multidimensional Convolution (Truncated)
#define PROBLEM "https://judge.yosupo.jp/problem/multivariate_convolution"
#pragma GCC optimize("O3,unroll-loops")
#define CP_ALGO_CHECKPOINT
#pragma GCC target("avx2")
#include <bits/stdc++.h>
//#include "blazingio/blazingio.min.hpp"
#include "cp-algo/math/multivar.hpp"

using namespace std;
using namespace cp_algo::math::fft;

const int mod = 998244353;
using base = cp_algo::math::modint<mod>;

void solve() {
    int k;
    cin >> k;
    cp_algo::big_vector<size_t> N(k);
    for(auto &n: N) cin >> n;
    multivar<base> a(N), b(N);
    a.read();
    b.read();
    a.mul(b);
    a.print();
    cp_algo::checkpoint<1>();
}

signed main() {
    //freopen("input.txt", "r", stdin);
    ios::sync_with_stdio(0);
    cin.tie(0);
    solve();
}

Traceback (most recent call last):
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/resolver.py", line 290, in resolve
    bundled_code = language.bundle(path, basedir=basedir)
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/languages/cplusplus.py", line 243, in bundle
    bundler.update(path)
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/languages/cplusplus_bundle.py", line 322, in update
    assert len(lines) == len(uncommented_lines)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AssertionError

Test cases

Env	Name	Status	Elapsed	Memory
g++	dim1_00	AC	61 ms	40 MB
g++	dim1_01	AC	65 ms	40 MB
g++	dim2_00	AC	79 ms	56 MB
g++	dim2_01	AC	80 ms	57 MB
g++	example_00	AC	2 ms	4 MB
g++	example_01	AC	2 ms	4 MB
g++	example_02	AC	2 ms	4 MB
g++	k0_00	AC	2 ms	4 MB
g++	k0_01	AC	2 ms	4 MB
g++	max_random_00	AC	157 ms	136 MB
g++	max_random_01	AC	166 ms	136 MB
g++	small_00	AC	3 ms	4 MB
g++	small_01	AC	3 ms	4 MB
g++	threes_00	AC	357 ms	251 MB
g++	threes_01	AC	309 ms	234 MB
g++	twos_00	AC	551 ms	319 MB
g++	twos_01	AC	243 ms	153 MB

Back to top page