結果
問題 | No.1391 ±1 Abs Sum |
ユーザー | Kiri8128 |
提出日時 | 2021-02-12 22:07:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,546 ms / 2,000 ms |
コード長 | 977 bytes |
コンパイル時間 | 263 ms |
コンパイル使用メモリ | 87,060 KB |
実行使用メモリ | 261,292 KB |
最終ジャッジ日時 | 2023-09-27 04:16:38 |
合計ジャッジ時間 | 28,230 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 73 ms
71,284 KB |
testcase_01 | AC | 73 ms
71,364 KB |
testcase_02 | AC | 71 ms
71,332 KB |
testcase_03 | AC | 72 ms
71,292 KB |
testcase_04 | AC | 72 ms
71,560 KB |
testcase_05 | AC | 72 ms
71,296 KB |
testcase_06 | AC | 72 ms
71,492 KB |
testcase_07 | AC | 76 ms
71,328 KB |
testcase_08 | AC | 83 ms
75,952 KB |
testcase_09 | AC | 81 ms
75,784 KB |
testcase_10 | AC | 80 ms
75,748 KB |
testcase_11 | AC | 960 ms
259,504 KB |
testcase_12 | AC | 805 ms
259,500 KB |
testcase_13 | AC | 826 ms
260,124 KB |
testcase_14 | AC | 567 ms
259,020 KB |
testcase_15 | AC | 586 ms
259,360 KB |
testcase_16 | AC | 774 ms
260,212 KB |
testcase_17 | AC | 613 ms
259,380 KB |
testcase_18 | AC | 799 ms
260,872 KB |
testcase_19 | AC | 655 ms
259,024 KB |
testcase_20 | AC | 888 ms
259,800 KB |
testcase_21 | AC | 1,266 ms
259,152 KB |
testcase_22 | AC | 1,355 ms
259,152 KB |
testcase_23 | AC | 1,546 ms
260,472 KB |
testcase_24 | AC | 1,422 ms
259,208 KB |
testcase_25 | AC | 1,543 ms
258,804 KB |
testcase_26 | AC | 1,543 ms
260,312 KB |
testcase_27 | AC | 1,053 ms
259,164 KB |
testcase_28 | AC | 1,124 ms
258,832 KB |
testcase_29 | AC | 1,330 ms
259,956 KB |
testcase_30 | AC | 1,120 ms
259,508 KB |
testcase_31 | AC | 875 ms
259,296 KB |
testcase_32 | AC | 521 ms
259,072 KB |
testcase_33 | AC | 675 ms
259,436 KB |
testcase_34 | AC | 994 ms
259,508 KB |
testcase_35 | AC | 71 ms
71,288 KB |
testcase_36 | AC | 662 ms
261,292 KB |
ソースコード
def f(k): b = A[k] L = sorted([abs(a - b) for a in A]) return sum(L[:K]) - sum(L[K:]) N, K = map(int, input().split()) A = [int(a) for a in input().split()] # 1 l, r = 0, N - 1 fl, fr = f(l), f(r) ans = min(fl, fr) while r - l > 2: m1 = (l * 2 + r) // 3 m2 = (l + r * 2) // 3 fm1, fm2 = f(m1), f(m2) if fm1 < fm2: r = m2 ans = min(ans, fm1) else: l = m1 ans = min(ans, fm2) if r - l > 1: ans = min(ans, f(l + r >> 1)) # 2 mi = 10 ** 60 mii = -1 s = max(1, N // 100) for i in range(0, N, s): fi = f(i) if fi < mi: mi = fi mii = i ans = min(ans, mi) l = max(0, mii - s) r = min(N, mii + s) while r - l > 3: m1 = (l * 2 + r) // 3 m2 = (l + r * 2) // 3 fm1, fm2 = f(m1), f(m2) if fm1 < fm2: r = m2 ans = min(ans, fm1) else: l = m1 ans = min(ans, fm2) if r - l > 1: for i in range(l + 1, r): ans = min(ans, f(i)) print(ans)