結果

問題 No.1391 ±1 Abs Sum
ユーザー aaaaaaaaaa2230
提出日時 2021-05-11 13:50:35
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 517 bytes
コンパイル時間 277 ms
コンパイル使用メモリ 82,092 KB
実行使用メモリ 259,536 KB
最終ジャッジ日時 2024-09-21 19:09:47
合計ジャッジ時間 20,013 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 13 WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff #

n,k = map(int,input().split())
A = list(map(int,input().split()))

def calc(x):
    count = []
    for a in A:
        count.append(abs(x-a))
    count.sort()
    ans = 0
    for i in range(n):
        if i < k:
            ans += count[i]
        else:
            ans -= count[i]
    return ans


l = A[0]-1
r = A[-1]+1
while r > l+1:
    m1 = (r+2*l)//3
    m2 = (2*r+l)//3
    c1 = calc(m1)
    c2 = calc(m2)
    if c1 <= c2:
        r = m2
    else:
        l = m2
ans = 10**20
print(min(calc(A[0]),calc(A[-1])))
0