結果

問題 No.931 Multiplicative Convolution
ユーザー lam6er
提出日時 2025-03-31 17:36:06
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 644 bytes
コンパイル時間 317 ms
コンパイル使用メモリ 83,072 KB
実行使用メモリ 68,664 KB
最終ジャッジ日時 2025-03-31 17:36:53
合計ジャッジ時間 7,556 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 4 TLE * 2 -- * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

def main():
    import sys
    input = sys.stdin.read().split()
    ptr = 0
    p = int(input[ptr])
    ptr += 1
    A = list(map(int, input[ptr:ptr + p-1]))
    ptr += p-1
    B = list(map(int, input[ptr:ptr + p-1]))
    
    # Precompute modular inverses for 1..p-1
    inv = [0] * p
    for i in range(1, p):
        inv[i] = pow(i, p-2, p)
    
    C = [0] * (p-1)
    for k in range(1, p):
        total = 0
        for i in range(1, p):
            j = (k * inv[i]) % p
            total = (total + A[i-1] * B[j-1]) % MOD
        C[k-1] = total
    
    print(' '.join(map(str, C)))

if __name__ == '__main__':
    main()
0