Sqrt of Power Series (Sparse) (verify/poly/sparse/sqrt.test.cpp)

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

Depends on

• cp-algo/math/affine.hpp
• cp-algo/math/combinatorics.hpp
• cp-algo/math/common.hpp
• cp-algo/math/cvector.hpp
• cp-algo/math/dft.hpp
• cp-algo/math/fft.hpp
• cp-algo/math/poly/base.hpp
• cp-algo/math/poly/sparse/impl.hpp
• cp-algo/math/poly/sparse/sqrt.hpp
• cp-algo/number_theory/discrete_sqrt.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 Sqrt of Power Series (Sparse)
#define PROBLEM "https://judge.yosupo.jp/problem/sqrt_of_formal_power_series_sparse"
#define CP_ALGO_MAXN (1 << 20)
#include <bits/stdc++.h>
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2")
#include "blazingio/blazingio.min.hpp"
#include "cp-algo/math/poly/sparse/sqrt.hpp"
using namespace std;
using namespace cp_algo::math;
using base = modint<998244353>;
using polyn = poly_t<base>;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n, k;
    cin >> n >> k;
    polyn::Vector a(n);
    for(int j = 0; j < k; j++) {
        int i;
        cin >> i;
        cin >> a[i];
    }
    auto q = sqrt_sparse(polyn(std::move(a)), n);
    if(q) {q->print(n);}
    else {cout << -1 << '\n';}
}

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	11 ms	10 MB
g++	example_01	AC	2 ms	6 MB
g++	example_02	AC	2 ms	6 MB
g++	max_random_00	AC	3 ms	7 MB
g++	max_random_01	AC	4 ms	7 MB
g++	max_random_02	AC	3 ms	7 MB
g++	max_random_03	AC	43 ms	20 MB
g++	max_random_04	AC	4 ms	7 MB
g++	max_random_05	AC	42 ms	20 MB
g++	max_random_06	AC	4 ms	7 MB
g++	max_random_07	AC	4 ms	7 MB
g++	max_random_08	AC	4 ms	7 MB
g++	max_random_09	AC	3 ms	7 MB
g++	min_K_00	AC	9 ms	6 MB
g++	min_K_01	AC	11 ms	6 MB
g++	random_00	AC	4 ms	6 MB
g++	random_01	AC	3 ms	6 MB
g++	random_02	AC	3 ms	6 MB
g++	random_03	AC	4 ms	6 MB
g++	random_04	AC	4 ms	7 MB
g++	random_05	AC	20 ms	11 MB
g++	random_06	AC	4 ms	6 MB
g++	random_07	AC	26 ms	12 MB
g++	random_08	AC	35 ms	17 MB
g++	random_09	AC	3 ms	6 MB
g++	small_N_00	AC	2 ms	6 MB
g++	small_N_01	AC	2 ms	6 MB
g++	small_N_02	AC	2 ms	6 MB
g++	small_N_03	AC	2 ms	6 MB
g++	small_N_04	AC	11 ms	10 MB
g++	small_N_05	AC	11 ms	10 MB
g++	small_N_06	AC	11 ms	10 MB
g++	small_N_07	AC	11 ms	10 MB
g++	small_N_08	AC	12 ms	10 MB
g++	small_N_09	AC	2 ms	6 MB
g++	small_dense_00	AC	58 ms	22 MB
g++	small_dense_01	AC	6 ms	7 MB
g++	small_dense_02	AC	5 ms	7 MB
g++	small_dense_03	AC	19 ms	7 MB
g++	small_dense_04	AC	6 ms	7 MB
g++	small_dense_05	AC	58 ms	22 MB
g++	small_dense_06	AC	56 ms	22 MB
g++	small_dense_07	AC	6 ms	7 MB
g++	small_dense_08	AC	60 ms	22 MB
g++	small_dense_09	AC	19 ms	7 MB

Back to top page