Composition of Formal Power Series (verify/poly/compose.test.cpp)

• View this file on GitHub
• Last update: 2026-09-10 02:45:56+02:00
• Problem: https://judge.yosupo.jp/problem/composition_of_formal_power_series

Depends on

• cp-algo/math/combinatorics.hpp
• cp-algo/math/common.hpp
• cp-algo/math/cvector.hpp
• cp-algo/math/dft.hpp
• cp-algo/math/factorials.hpp
• cp-algo/math/fft.hpp
• cp-algo/math/poly/base.hpp
• cp-algo/math/poly/calculus.hpp
• cp-algo/math/poly/compose.hpp
• cp-algo/math/poly/series.hpp
• cp-algo/math/poly/series/exp.hpp
• cp-algo/math/poly/series/inv.hpp
• cp-algo/math/poly/series/log.hpp
• cp-algo/math/poly/series/pow.hpp
• cp-algo/math/poly/transform.hpp
• cp-algo/number_theory/modint.hpp
• cp-algo/random/rng.hpp
• cp-algo/util/big_alloc.hpp
• cp-algo/util/bump_alloc.hpp
• cp-algo/util/checkpoint.hpp
• cp-algo/util/complex.hpp
• cp-algo/util/simd.hpp

Code

// @brief Composition of Formal Power Series
#define PROBLEM "https://judge.yosupo.jp/problem/composition_of_formal_power_series"
#include <bits/stdc++.h>
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2")
#include "blazingio/blazingio.min.hpp"
#include "cp-algo/math/poly/compose.hpp"

using namespace std;
using namespace cp_algo::math;

const int mod = 998244353;
using base = modint<mod>;
using polyn = poly_t<base>;

void solve() {
    int n;
    int64_t m;
    cin >> n;
    polyn::Vector a(n), b(n);
    for(auto &it: a) {cin >> it;}
    for(auto &it: b) {cin >> it;}
    compose(polyn(std::move(a)), polyn(std::move(b)), n).print(n);
}

signed main() {
    //freopen("input.txt", "r", stdin);
    ios::sync_with_stdio(0);
    cin.tie(0);
    int t = 1;
    while(t--) {
        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++	example_00	AC	8 ms	8 MB
g++	hack2_00	AC	262 ms	14 MB
g++	hack2_01	AC	215 ms	14 MB
g++	hack2_02	AC	225 ms	14 MB
g++	hack_00	AC	57 ms	7 MB
g++	hack_01	AC	5 ms	6 MB
g++	hack_02	AC	7 ms	7 MB
g++	max_random_00	AC	272 ms	14 MB
g++	max_random_01	AC	272 ms	14 MB
g++	max_random_02	AC	255 ms	14 MB
g++	max_random_03	AC	266 ms	14 MB
g++	max_random_04	AC	260 ms	14 MB
g++	random_00	AC	130 ms	11 MB
g++	random_01	AC	116 ms	11 MB
g++	random_02	AC	91 ms	7 MB
g++	random_03	AC	51 ms	6 MB
g++	random_04	AC	208 ms	13 MB
g++	small_00	AC	8 ms	8 MB
g++	small_01	AC	8 ms	8 MB
g++	small_02	AC	7 ms	8 MB
g++	small_03	AC	2 ms	6 MB
g++	small_04	AC	7 ms	8 MB
g++	small_05	AC	7 ms	8 MB
g++	small_06	AC	7 ms	8 MB
g++	small_07	AC	7 ms	8 MB
g++	small_08	AC	7 ms	8 MB
g++	small_09	AC	2 ms	6 MB

Back to top page