結果

問題 No.1068 #いろいろな色 / Red and Blue and more various colors (Hard)
ユーザー tktk_snsn
提出日時 2022-04-30 10:49:44
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,359 bytes
コンパイル時間 201 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 251,372 KB
最終ジャッジ日時 2024-06-29 16:30:23
合計ジャッジ時間 8,948 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 TLE * 1 -- * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

"""
ref
https://atcoder.jp/contests/typical90/submissions/29013313
"""
MOD = 998244353
RT = 3


def pr(n):
    return pow(RT, (MOD-1) // n, MOD)


def inv(x):
    return pow(x, MOD-2, MOD)


def FFT(a, w):
    N = len(a)
    if N == 1:
        return a
    even = a[::2]
    odd = a[1::2]
    ww = w * w % MOD
    EVEN = FFT(even, (ww) % MOD)
    ODD = FFT(odd, (ww) % MOD)

    A = [0] * N
    wk = 1
    for k in range(N):
        A[k] = (EVEN[k % (N//2)] + wk * ODD[k % (N//2)] % MOD) % MOD
        wk = wk * w % MOD
    return A


def convolution(a, b):
    d = len(a) + len(b) - 1
    n = 1
    while (1 << n) < d:
        n += 1
    N = 1 << n
    w = pr(N)
    a += [0] * (N - len(a))
    b += [0] * (N - len(b))
    A = FFT(a, w)
    B = FFT(b, w)
    C = [A[i] * B[i] % MOD for i in range(N)]

    winv = inv(w)
    c = FFT(C, winv)
    invN = inv(N)
    for i in range(N):
        c[i] = c[i] * invN % MOD
    return c[:d]


def main():
    N, Q = map(int, input().split())
    A = list(map(int, input().split()))
    B = list(map(int, input().split()))

    que = [[a-1, 1] for a in A]
    while len(que) > 1:
        nque = []
        while len(que) > 1:
            nque.append(convolution(que.pop(), que.pop()))
        if que:
            nque.append(que.pop())
        que = nque

    F = que.pop()
    for b in B:
        print(F[b])


main()
0