結果

問題 No.1391 ±1 Abs Sum
ユーザー H3PO4
提出日時 2021-02-12 22:15:10
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 534 ms / 2,000 ms
コード長 500 bytes
コンパイル時間 97 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 33,144 KB
最終ジャッジ日時 2024-07-19 22:21:17
合計ジャッジ時間 11,897 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

N, K = map(int, input().split())
A = tuple(map(int, input().split()))

l, r = 0, K
A0 = A[0]
plus, minus = sum(a - A0 for a in A[:K]), sum(a - A0 for a in A[K:])
ans = plus - minus
for i in range(1, N):
    x = A[i]
    diff = A[i] - A[i - 1]
    plus += diff * (2 * i - l - r)
    minus += diff * (l + r - N)
    while r < N and (arx := (A[r] - x)) < (alx := (x - A[l])):
        plus += arx - alx
        minus += alx - arx
        r += 1
        l += 1
    ans = min(plus - minus, ans)
print(ans)
0