結果

問題 No.1068 #いろいろな色 / Red and Blue and more various colors (Hard)
ユーザー tktk_snsntktk_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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
57,472 KB
testcase_01 AC 37 ms
52,352 KB
testcase_02 AC 38 ms
52,608 KB
testcase_03 AC 260 ms
82,176 KB
testcase_04 AC 217 ms
80,272 KB
testcase_05 AC 222 ms
80,384 KB
testcase_06 AC 197 ms
79,088 KB
testcase_07 AC 190 ms
79,744 KB
testcase_08 AC 220 ms
80,256 KB
testcase_09 AC 235 ms
80,432 KB
testcase_10 AC 144 ms
77,312 KB
testcase_11 AC 189 ms
79,132 KB
testcase_12 AC 138 ms
77,084 KB
testcase_13 TLE -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
権限があれば一括ダウンロードができます

ソースコード

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