TestCase/unittest/gcd_convolve.unittest.test.py

• View this file on GitHub
• Last update: 2021-06-24 01:16:50+09:00
• Problem: http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=ITP1_1_A

Depends on

• 添字 gcd による畳み込み (NumberTheory/Convolution/gcd_convolve.py)

Code

# verification-helper: PROBLEM http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=ITP1_1_A
import sys
input = sys.stdin.buffer.readline

from NumberTheory.Convolution.gcd_convolve import gcd_convolve


def naive_gcd_convolve(a, b):
    """O(N^2) naive calculation"""
    def gcd(x, y):
        while y:
            x, y = y, x % y
        return x

    n = min(len(a), len(b))
    res = [0] * n
    for i, va in enumerate(a[1:], 1):
        for j, vb in enumerate(b[1:], 1):
            res[gcd(i, j)] += va * vb
    return res


def main():
    n = 1000
    a = [1] * n
    b = [1] * n
    a[0] = 0
    b[0] = 0
    assert(gcd_convolve(a, b) == naive_gcd_convolve(a, b))


if __name__ == '__main__':
    main()
    print("Hello World")

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.12.4/x64/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 "/opt/hostedtoolcache/Python/3.12.4/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/python.py", line 96, in bundle
    raise NotImplementedError
NotImplementedError

Back to top page