結果
| 問題 |
No.738 平らな農地
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 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 |
ソースコード
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)