結果

問題 No.1618 Convolution?
ユーザー lam6er
提出日時 2025-03-31 17:25:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 229 ms / 2,000 ms
コード長 1,008 bytes
コンパイル時間 266 ms
コンパイル使用メモリ 82,648 KB
実行使用メモリ 147,176 KB
最終ジャッジ日時 2025-03-31 17:27:05
合計ジャッジ時間 8,094 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

# Adjust to 1-based indexing by prepending a 0
A = [0] + A
B = [0] + B

# Precompute prefix sums for A and B arrays
prefix_A = [0] * (n + 1)
prefix_jA = [0] * (n + 1)
prefix_B = [0] * (n + 1)
prefix_iB = [0] * (n + 1)

for i in range(1, n + 1):
    prefix_A[i] = prefix_A[i - 1] + A[i]
    prefix_jA[i] = prefix_jA[i - 1] + i * A[i]
    prefix_B[i] = prefix_B[i - 1] + B[i]
    prefix_iB[i] = prefix_iB[i - 1] + i * B[i]

result = []
for k in range(1, 2 * n + 1):
    a = max(1, k - n)
    b = min(n, k - 1)
    if a > b:
        sum_A = 0
        sum_jA_val = 0
        sum_B = 0
        sum_iB_val = 0
    else:
        sum_A = prefix_A[b] - prefix_A[a - 1]
        sum_jA_val = prefix_jA[b] - prefix_jA[a - 1]
        sum_B = prefix_B[b] - prefix_B[a - 1]
        sum_iB_val = prefix_iB[b] - prefix_iB[a - 1]
    c = k * (sum_A + sum_B) - (sum_jA_val + sum_iB_val)
    result.append(str(c))

print(' '.join(result))
0