結果

問題 No.1391 ±1 Abs Sum
コンテスト
ユーザー zeronosu77108
提出日時 2021-02-12 22:57:39
言語 Kotlin
(2.3.20)
コンパイル:
kotlinc _filename_ -include-runtime -d main.jar
実行:
kotlin main.jar
結果
WA  
実行時間 -
コード長 739 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 8,466 ms
コンパイル使用メモリ 530,352 KB
実行使用メモリ 123,352 KB
最終ジャッジ日時 2026-08-19 05:19:37
合計ジャッジ時間 47,727 ms
ジャッジサーバーID
(参考情報)
judge1_0 / judge3_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 13 WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

import kotlin.collections.*
import kotlin.math.*

@kotlin.ExperimentalStdlibApi
fun main() {
    val (n, k) = readLine()!!.split(" ").map { it.toInt() }
    val a = readLine()!!.split(" ").map { it.toLong() }.toTypedArray().sorted()

    val amin = a.min()!!
    val amax = a.max()!!

    fun f(ai : Long) : Long {
        val l = a.map { abs(ai - it) }.sorted()
        return l.take(k).sum() - l.takeLast(n-k).sum()
    }

    var l = 0; var r = n - 1
    while(l + 10 < r) {
        val ll = (l+l+r) / 3
        val rr = (l+r+r) / 3

        if (f(a[ll]) < f(a[rr])) l = ll
        else r = rr

    }
    
    var ans = minOf(f(a.first()), f(a.last()))
    for (i in l .. r) {
        ans = minOf(ans, f(a[i]))
    }
    println(ans)
}
0