結果
| 問題 | 
                            No.738 平らな農地
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2018-09-28 21:48:59 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                TLE
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 528 bytes | 
| コンパイル時間 | 259 ms | 
| コンパイル使用メモリ | 82,384 KB | 
| 実行使用メモリ | 94,804 KB | 
| 最終ジャッジ日時 | 2024-10-12 05:51:20 | 
| 合計ジャッジ時間 | 10,853 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge3 / judge2 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 5 | 
| other | AC * 6 WA * 3 TLE * 2 -- * 76 | 
ソースコード
from collections import defaultdict
N, K = map(int, input().split())
A = list(map(int, input().split()))
ans = 1e15
for i in range(N - K + 1):
    lo, hi = 1, 10**9
    while lo <= hi:
        mi = (lo + hi + 1)//2
        #print(i, lo, hi)
        t_lo = sum(abs(A[j] - lo) for j in range(i, i + K))
        t_hi = sum(abs(A[j] - hi) for j in range(i, i + K))
        if t_lo < t_hi:
            hi = mi - 1
            ans = min(ans, t_lo)
        else:
            lo = mi + 1
            ans = min(ans, t_hi)
print(ans)