結果

問題 No.754 畳み込みの和
コンテスト
ユーザー lam6er
提出日時 2025-03-20 21:06:02
言語 PyPy3
(8.0.0 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
WJ  
(最新)
AC  
(最初)
実行時間 -
コード長 644 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
最終ジャッジ日時 2026-07-07 12:47:57
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_1
このコードへのチャレンジ
（要ログイン）

ソースコード

diff #
raw source code 

import sys

MOD = 10**9 + 7

def main():
    data = list(map(int, sys.stdin.read().split()))
    ptr = 0
    n = data[ptr]
    ptr += 1
    a = data[ptr:ptr + n + 1]
    ptr += n + 1
    b = data[ptr:ptr + n + 1]
    ptr += n + 1
    
    if n == 0:
        print((a[0] * b[0]) % MOD)
        return
    
    # Compute prefix sum of b
    s = [0] * (n + 1)
    s[0] = b[0] % MOD
    for i in range(1, n + 1):
        s[i] = (s[i - 1] + b[i]) % MOD
    
    total = 0
    for p in range(n + 1):
        k = n - p
        sum_b = s[k]
        total = (total + a[p] * sum_b) % MOD
    
    print(total % MOD)

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