結果
| 問題 | 
                            No.738 平らな農地
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2018-09-28 21:52:06 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                TLE
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 542 bytes | 
| コンパイル時間 | 170 ms | 
| コンパイル使用メモリ | 82,524 KB | 
| 実行使用メモリ | 84,648 KB | 
| 最終ジャッジ日時 | 2024-10-12 06:12:16 | 
| 合計ジャッジ時間 | 5,791 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge1 / judge3 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 5 | 
| other | AC * 2 TLE * 1 -- * 84 | 
ソースコード
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:
        l = (2*lo + hi)//3
        h = (lo + 2*hi)//3
        #print(i, lo, hi)
        s_l = sum(abs(A[j] - l) for j in range(i, i + K))
        s_h = sum(abs(A[j] - h) for j in range(i, i + K))
        if s_l < s_h:
            hi = h - 1
            ans = min(ans, s_l)
        else:
            lo = l + 1
            ans = min(ans, s_h)
print(ans)