結果

問題 No.1618 Convolution?
ユーザー 👑 Kazun
提出日時 2021-12-15 05:03:44
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 247 ms / 2,000 ms
コード長 649 bytes
コンパイル時間 201 ms
コンパイル使用メモリ 82,116 KB
実行使用メモリ 160,008 KB
最終ジャッジ日時 2024-07-23 19:52:29
合計ジャッジ時間 9,651 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

def accumulate(X):
    Y=[0]*(len(X))
    for i in range(1,len(X)):
        Y[i]=X[i]+Y[i-1]
    return Y

#==================================================
N=int(input())
A=[0]+list(map(int,input().split()))+[0]*N
B=[0]+list(map(int,input().split()))+[0]*N

E=[i*A[i] for i in range(2*N+1)]
F=[j*B[j] for j in range(2*N+1)]

#==================================================
A_cum=accumulate(A)
B_cum=accumulate(B)
E_cum=accumulate(E)
F_cum=accumulate(F)

C=[0]*(2*N+1)
for k in range(2,2*N+1):
    l=max(1,k-N); r=min(k-1,N)
    C[k]=k*((A_cum[r]-A_cum[l-1])+(B_cum[r]-B_cum[l-1]))-((E_cum[r]-E_cum[l-1])+(F_cum[r]-F_cum[l-1]))

print(*C[1:])
0