Polynomial Interpolation (Geometric Sequence) (verify/poly/inter_geo.test.cpp)

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

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/fft.hpp
• cp-algo/math/poly.hpp
• cp-algo/math/poly/impl/div.hpp
• cp-algo/math/poly/impl/euclid.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 Polynomial Interpolation (Geometric Sequence)
#define PROBLEM "https://judge.yosupo.jp/problem/polynomial_interpolation_on_geometric_sequence"
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2")
#include "cp-algo/math/poly.hpp"
#include <bits/stdc++.h>

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, a, r;
    cin >> n >> a >> r;
    polyn::Vector y(n);
    copy_n(istream_iterator<base>(cin), n, begin(y));
    polyn(y).chirpz_inverse(r, n).mulx(base(a).inv()).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	3 ms	4 MB
g++	example_01	AC	2 ms	4 MB
g++	max_random_00	AC	202 ms	53 MB
g++	max_random_01	AC	208 ms	53 MB
g++	max_random_02	AC	209 ms	53 MB
g++	nar_0_00	AC	3 ms	4 MB
g++	nar_0_01	AC	2 ms	4 MB
g++	nar_0_02	AC	2 ms	4 MB
g++	nar_0_03	AC	2 ms	4 MB
g++	nar_0_04	AC	2 ms	4 MB
g++	nar_0_05	AC	2 ms	4 MB
g++	nar_0_06	AC	2 ms	4 MB
g++	nar_0_07	AC	2 ms	4 MB
g++	near_pow2_00	AC	99 ms	31 MB
g++	near_pow2_01	AC	101 ms	31 MB
g++	near_pow2_02	AC	109 ms	31 MB
g++	near_pow2_03	AC	140 ms	47 MB
g++	near_pow2_04	AC	139 ms	47 MB
g++	pow_rN_equal_1_00	AC	138 ms	48 MB
g++	pow_rN_equal_1_01	AC	172 ms	55 MB
g++	pow_rN_equal_1_02	AC	24 ms	11 MB
g++	random_00	AC	162 ms	52 MB
g++	random_01	AC	177 ms	55 MB
g++	random_02	AC	27 ms	12 MB
g++	small_00	AC	2 ms	4 MB
g++	small_01	AC	2 ms	4 MB
g++	small_02	AC	2 ms	4 MB
g++	y0_00	AC	45 ms	8 MB

Back to top page