結果

問題 No.1618 Convolution?
ユーザー tktk_snsn
提出日時 2022-05-17 07:04:42
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,208 ms / 2,000 ms
コード長 446 bytes
コンパイル時間 357 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 71,056 KB
最終ジャッジ日時 2024-09-15 00:33:30
合計ジャッジ時間 20,931 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #

N = int(input())
A = list(map(int, input().split()))
B = list(map(int, input().split()))

SA = [0] * (N + 1)
SB = [0] * (N + 1)
for i, a in enumerate(A, 1):
    SA[i] = SA[i-1] + a
for i, b in enumerate(B, 1):
    SB[i] = SB[i-1] + b

C = [0] * (2 * N + 1)
X = 0
Y = 0
for i in range(2, 2 * N + 1):
    L = max(0, i - N - 1)
    R = min(N, i - 1)
    X += SA[R] - SA[L]
    Y += SB[R] - SB[L]
    C[i] = X + Y - N * (SA[L] + SB[L])

print(*C[1:])
0