verify/AI/ntt_test.nim

View this file on GitHub
Last update: 2026-09-04 10:21:15+09:00
Problem: https://onlinejudge.u-aizu.ac.jp/problems/ITP1_1_A

Depends on

Code

# verification-helper: PROBLEM https://onlinejudge.u-aizu.ac.jp/problems/ITP1_1_A
echo "Hello World"

import sequtils
import cplib/convolution/ntt
import cplib/modint/modint

block static_barrett_ntt:
  type Mint = modint998244353_barrett
  var f = @[Mint(1), Mint(2), Mint(3), Mint(4)]
  ntt(f)
  intt(f)
  assert f.mapIt(it.val) == @[1, 2, 3, 4]

block static_montgomery_ntt:
  type Mint = modint998244353_montgomery
  var f = @[Mint(5), Mint(6), Mint(7), Mint(8)]
  ntt(f)
  intt(f)
  assert f.mapIt(it.val) == @[5, 6, 7, 8]

block dynamic_barrett_ntt:
  type Mint = modint_barrett
  Mint.setMod(998244353)
  var f = @[Mint(9), Mint(10)]
  ntt(f)
  intt(f)
  assert f.mapIt(it.val) == @[9, 10]

Traceback (most recent call last):
  File "/home/runner/.local/lib/python3.12/site-packages/onlinejudge_verify/documentation/build.py", line 71, in _render_source_code_stat
    bundled_code = language.bundle(stat.path, basedir=basedir, options={'include_paths': [basedir]}).decode()
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/runner/.local/lib/python3.12/site-packages/onlinejudge_verify/languages/nim.py", line 86, in bundle
    raise NotImplementedError
NotImplementedError