結果

問題 No.1618 Convolution?
ユーザー ayaoniayaoni
提出日時 2021-07-23 00:13:46
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,432 ms / 2,000 ms
コード長 3,965 bytes
コンパイル時間 2,133 ms
コンパイル使用メモリ 86,260 KB
実行使用メモリ 161,768 KB
最終ジャッジ日時 2023-09-24 21:05:16
合計ジャッジ時間 23,504 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 70 ms
71,112 KB
testcase_01 AC 71 ms
71,288 KB
testcase_02 AC 1,368 ms
153,216 KB
testcase_03 AC 1,392 ms
161,768 KB
testcase_04 AC 798 ms
129,644 KB
testcase_05 AC 198 ms
81,500 KB
testcase_06 AC 1,370 ms
148,864 KB
testcase_07 AC 1,373 ms
149,972 KB
testcase_08 AC 805 ms
131,268 KB
testcase_09 AC 1,430 ms
159,396 KB
testcase_10 AC 1,331 ms
144,844 KB
testcase_11 AC 1,410 ms
155,016 KB
testcase_12 AC 1,428 ms
160,216 KB
testcase_13 AC 1,426 ms
160,020 KB
testcase_14 AC 1,426 ms
160,996 KB
testcase_15 AC 1,432 ms
160,024 KB
testcase_16 AC 1,424 ms
160,680 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

sys.setrecursionlimit(10**7)
def I(): return int(sys.stdin.readline().rstrip())
def MI(): return map(int,sys.stdin.readline().rstrip().split())
def LI(): return list(map(int,sys.stdin.readline().rstrip().split()))
def LI2(): return list(map(int,sys.stdin.readline().rstrip()))
def S(): return sys.stdin.readline().rstrip()
def LS(): return list(sys.stdin.readline().rstrip().split())
def LS2(): return list(sys.stdin.readline().rstrip())


# 998244353 = 119*2**23+1

mod = 998244353
primitive_root = 3  # mod の原始根
roots = [pow(primitive_root,(mod-1) >> i,mod) for i in range(24)]
inv_roots = [pow(r,mod-2,mod) for r in roots]
# roots[i] = 1 の 2**i 乗根、inv_roots[i] = 1 の 2**i 乗根の逆元


def ntt(A,n):
    for i in range(n):
        m = 1 << (n-i-1)
        for start in range(1 << i):
            w = 1
            start *= m*2
            for j in range(m):
                A[start+j],A[start+j+m] = (A[start+j]+A[start+j+m]) % mod,(A[start+j]-A[start+j+m])*w % mod
                w *= roots[n-i]
                w %= mod
    return A


def inv_ntt(A,n):
    for i in range(n):
        m = 1 << i
        for start in range(1 << (n-i-1)):
            w = 1
            start *= m*2
            for j in range(m):
                A[start+j],A[start+j+m] = (A[start+j]+A[start+j+m]*w) % mod,(A[start+j]-A[start+j+m]*w) % mod
                w *= inv_roots[i+1]
                w %= mod
    a = pow(2,n*(mod-2),mod)
    for i in range(1 << n):
        A[i] *= a
        A[i] %= mod
    return A


def convolution(A,B):
    a,b = len(A),len(B)
    deg = a+b-2
    n = deg.bit_length()
    N = 1 << n
    A += [0]*(N-a)  # A の次数を 2冪-1 にする
    B += [0]*(N-b)  # B の次数を 2冪-1 にする
    A = ntt(A,n)
    B = ntt(B,n)
    C = [(A[i]*B[i]) % mod for i in range(N)]
    C = inv_ntt(C,n)
    return C[:deg+1]


N = I()
A = [0]+LI()
B = [0]+LI()
AB = [a+b for a,b in zip(A,B)]
X = [i for i in range(N+1)]
ANS0 = convolution(AB[:],X[:])


# 1012924417 = 483*2**21+1

mod = 1012924417
primitive_root = 5  # mod の原始根
roots = [pow(primitive_root,(mod-1) >> i,mod) for i in range(22)]
inv_roots = [pow(r,mod-2,mod) for r in roots]
# roots[i] = 1 の 2**i 乗根、inv_roots[i] = 1 の 2**i 乗根の逆元


def ntt(A,n):
    for i in range(n):
        m = 1 << (n-i-1)
        for start in range(1 << i):
            w = 1
            start *= m*2
            for j in range(m):
                A[start+j],A[start+j+m] = (A[start+j]+A[start+j+m]) % mod,(A[start+j]-A[start+j+m])*w % mod
                w *= roots[n-i]
                w %= mod
    return A


def inv_ntt(A,n):
    for i in range(n):
        m = 1 << i
        for start in range(1 << (n-i-1)):
            w = 1
            start *= m*2
            for j in range(m):
                A[start+j],A[start+j+m] = (A[start+j]+A[start+j+m]*w) % mod,(A[start+j]-A[start+j+m]*w) % mod
                w *= inv_roots[i+1]
                w %= mod
    a = pow(2,n*(mod-2),mod)
    for i in range(1 << n):
        A[i] *= a
        A[i] %= mod
    return A


def convolution(A,B):
    a,b = len(A),len(B)
    deg = a+b-2
    n = deg.bit_length()
    N = 1 << n
    A += [0]*(N-a)  # A の次数を 2冪-1 にする
    B += [0]*(N-b)  # B の次数を 2冪-1 にする
    A = ntt(A,n)
    B = ntt(B,n)
    C = [(A[i]*B[i]) % mod for i in range(N)]
    C = inv_ntt(C,n)
    return C[:deg+1]


ANS1 = convolution(AB[:],X[:])


def ext_gcd(a,b):  # gcd(a,b) と a*x+b*y == gcd(a,b) の整数解(x,y)
    if b > 0:
        d,x,y = ext_gcd(b,a % b)
        return d,y,x-(a//b)*y
    return a,1,0


def remainder(V):  # Z == Xi (mod Yi) たる Z, lcm(Yi)
    x,lcm = 0,1
    for X,Y in V:
        g,a,b = ext_gcd(lcm,Y)
        x,lcm = (Y*b*x+lcm*a*X)//g,lcm*(Y//g)
        x %= lcm
    return x,lcm


ANS = []
for i in range(1,2*N+1):
    a0 = ANS0[i]
    a1 = ANS1[i]
    a,l = remainder([(a0,998244353),(a1,1012924417)])
    ANS.append(a)

print(*ANS)
0