結果
| 問題 | 
                            No.738 平らな農地
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2018-09-28 22:05:10 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                WA
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 676 bytes | 
| コンパイル時間 | 173 ms | 
| コンパイル使用メモリ | 82,176 KB | 
| 実行使用メモリ | 91,136 KB | 
| 最終ジャッジ日時 | 2024-10-12 06:20:51 | 
| 合計ジャッジ時間 | 22,507 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge4 / judge3 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 5 | 
| other | AC * 81 WA * 6 | 
ソースコード
from collections import defaultdict
N, K = map(int, input().split())
A = list(map(int, input().split()))
ans = 1e15
lo, hi = 1, 10**9
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)