結果
| 問題 | 
                            No.1391 ±1 Abs Sum
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 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 | 
ソースコード
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])))