結果

問題 No.738 平らな農地
ユーザー roiti46
提出日時 2018-09-28 22:11:10
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 682 bytes
コンパイル時間 228 ms
コンパイル使用メモリ 82,452 KB
実行使用メモリ 91,192 KB
最終ジャッジ日時 2024-10-12 06:22:27
合計ジャッジ時間 23,686 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 80 WA * 7
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict

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

ans = 1e17
lo, hi = min(A), max(A)
while lo <= hi:
    l = (2*lo + hi)//3
    h = (lo + 2*hi)//3
    s_l = min_l = sum(abs(A[j] - l) for j in range(K))
    for j in range(K, N):
        s_l = s_l - abs(A[j - K] - l) + abs(A[j] - l)
        min_l = min(min_l, s_l)
    s_h = min_h = sum(abs(A[j] - h) for j in range(K))
    for j in range(K, N):
        s_h = s_h - abs(A[j - K] - h) + abs(A[j] - h)
        min_h = min(min_h, s_h)

    if min_l < min_h:
        hi = h - 1
        ans = min(ans, min_l)
    else:
        lo = l + 1
        ans = min(ans, min_h)
print(ans) 

0