expansion/$tests/convolution/or_convolution.test.py

View this file on GitHub
Last update: 2023-07-23 08:45:17+09:00
Problem: https://judge.yosupo.jp/problem/bitwise_and_convolution
Link: View error logs on GitHub Actions

Code

# verification-helper: PROBLEM https://judge.yosupo.jp/problem/bitwise_and_convolution
from pathlib import Path
import sys

sys.path.append(str(Path(__file__).resolve().parent.parent.parent.parent))


def or_convolution(A, B, MOD=-1):
    n = max(len(A), len(B))
    l = (n - 1).bit_length()
    n = 1 << l
    A += [0] * (n - len(A))
    B += [0] * (n - len(B))

    def f(A):
        for i in range(l):
            for bit in range(n):
                if bit >> i & 1:
                    A[bit] += A[bit ^ (1 << i)]
                    if MOD != -1:
                        A[bit] %= MOD

    def invf(A):
        for i in range(l):
            for bit in range(n):
                if bit >> i & 1:
                    A[bit] -= A[bit ^ (1 << i)]
                    if MOD != -1:
                        A[bit] %= MOD

    f(A)
    f(B)
    if MOD != -1:
        C = [a * b % MOD for a, b in zip(A, B)]
    else:
        C = [a * b for a, b in zip(A, B)]
    invf(C)
    return C


n = int(input())
A = list(map(int, input().split()))
B = list(map(int, input().split()))
A_ = [0] * (1 << n)
B_ = [0] * (1 << n)
mask = (1 << n) - 1
for i in range(1 << n):
    A_[i ^ mask] = A[i]
    B_[i ^ mask] = B[i]
C_ = or_convolution(A_, B_, 998244353)
C = [0] * (1 << n)
for i in range(1 << n):
    C[i ^ mask] = C_[i]
print(*C)

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.11.4/x64/lib/python3.11/site-packages/onlinejudge_verify/documentation/build.py", line 81, in _render_source_code_stat
    bundled_code = language.bundle(
                   ^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.11.4/x64/lib/python3.11/site-packages/onlinejudge_verify/languages/python.py", line 108, in bundle
    raise NotImplementedError
NotImplementedError