結果

問題 No.1618 Convolution?
ユーザー tamato
提出日時 2021-07-22 21:56:28
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 228 ms / 2,000 ms
コード長 614 bytes
コンパイル時間 214 ms
コンパイル使用メモリ 82,356 KB
実行使用メモリ 147,592 KB
最終ジャッジ日時 2024-07-17 17:39:55
合計ジャッジ時間 6,834 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #

mod = 1000000007
eps = 10**-9


def main():
    import sys
    input = sys.stdin.readline

    N = int(input())
    A = list(map(int, input().split()))
    B = list(map(int, input().split()))
    AB = [a + b for a, b in zip(A, B)]

    C = [0] * (2*N)
    imos = [0] * (2*N+2)
    imos2 = [0] * (2*N+2)
    for i, ab in enumerate(AB):
        imos2[i+1] += ab
        imos2[i+1 + N] += -ab * (N+1)
        imos2[i+2 + N] += ab * N
    for i in range(1, 2*N+1):
        imos[i] = imos[i-1] + imos2[i]
    for i in range(1, 2*N):
        C[i] = C[i-1] + imos[i]
    print(*C)


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