結果

問題 No.837 Noelちゃんと星々2
ユーザー FromBooskaFromBooska
提出日時 2023-06-03 18:46:05
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 100 ms / 2,000 ms
コード長 1,920 bytes
コンパイル時間 169 ms
コンパイル使用メモリ 82,328 KB
実行使用メモリ 101,104 KB
最終ジャッジ日時 2024-06-09 03:43:23
合計ジャッジ時間 3,284 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 98 ms
100,896 KB
testcase_01 AC 97 ms
101,000 KB
testcase_02 AC 97 ms
101,104 KB
testcase_03 AC 100 ms
100,800 KB
testcase_04 AC 96 ms
101,100 KB
testcase_05 AC 99 ms
100,680 KB
testcase_06 AC 97 ms
100,944 KB
testcase_07 AC 97 ms
100,880 KB
testcase_08 AC 95 ms
100,840 KB
testcase_09 AC 33 ms
54,156 KB
testcase_10 AC 33 ms
52,960 KB
testcase_11 AC 30 ms
53,968 KB
testcase_12 AC 30 ms
54,084 KB
testcase_13 AC 31 ms
52,708 KB
testcase_14 AC 31 ms
52,604 KB
testcase_15 AC 30 ms
53,260 KB
testcase_16 AC 30 ms
52,888 KB
testcase_17 AC 29 ms
53,132 KB
testcase_18 AC 31 ms
52,472 KB
testcase_19 AC 31 ms
53,384 KB
testcase_20 AC 30 ms
53,560 KB
testcase_21 AC 30 ms
52,264 KB
testcase_22 AC 31 ms
52,540 KB
testcase_23 AC 30 ms
53,944 KB
testcase_24 AC 29 ms
52,816 KB
testcase_25 AC 29 ms
53,484 KB
testcase_26 AC 30 ms
53,356 KB
testcase_27 AC 30 ms
53,868 KB
testcase_28 AC 31 ms
52,632 KB
testcase_29 AC 30 ms
53,544 KB
testcase_30 AC 30 ms
52,960 KB
testcase_31 AC 30 ms
52,344 KB
testcase_32 AC 31 ms
52,696 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

# まずYを昇順ソート
# 2個のmedian, 中央値を作ってそこに寄せればいい
# どこからを2グループ目とするかで全探索
# medianまでの距離の和は、median後の要素の和 - median前の要素の和
# つまり累積和で高速化できる、はず

N = int(input())
Y = list(map(int, input().split()))
Y.sort()

cumu = [0]
temp = 0
for i in range(N):
    temp += Y[i]
    cumu.append(temp)
    
ans = 10**20
for second in range(1, N):
    #print(second, Y[second])
    first_len = second
    second_len = N - first_len
    first_sum = cumu[second]
    second_sum = cumu[N]-first_sum
    #print(first_len, second_len, first_sum, second_sum)
    if first_len%2 == 0:
        first_median = (Y[first_len//2-1]+Y[first_len//2])//2
    else:
        first_median = Y[first_len//2]
    if second_len%2 == 0:
        second_median = (Y[first_len+second_len//2-1]+Y[first_len+second_len//2])//2
    else:
        second_median = Y[first_len+second_len//2]
    #print('first_median', first_median, 'second_median', second_median)
    # medianまでの距離の和は、median後の要素の和 - median前の要素の和
    if first_len%2 == 0:
        first_distance = (cumu[first_len]-cumu[(first_len+1)//2])-(cumu[(first_len+1)//2]-cumu[0])
    else:
        first_distance = (cumu[first_len]-cumu[(first_len+1)//2])-(cumu[(first_len+1)//2-1]-cumu[0])
    if second_len%2 == 0:
        second_distance = (cumu[first_len+second_len]-cumu[(second_len+1)//2+first_len])-(cumu[(second_len+1)//2+first_len]-cumu[0+first_len])
    else:
        second_distance = (cumu[first_len+second_len]-cumu[(second_len+1)//2+first_len])-(cumu[(second_len+1)//2-1+first_len]-cumu[0+first_len])
    ans = min(ans, first_distance+second_distance)

if Y[0] == Y[N-1]:
    # 全部おなじときはグループ1を1要素にして違うmedianにすれば1が達成可能
    print(1)
else:
    print(ans)
0