結果

問題 No.837 Noelちゃんと星々2
コンテスト
ユーザー FromBooska
提出日時 2023-06-03 18:46:05
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 82 ms / 2,000 ms
コード長 1,920 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 208 ms
コンパイル使用メモリ 84,480 KB
実行使用メモリ 105,600 KB
最終ジャッジ日時 2026-06-02 23:29:29
合計ジャッジ時間 3,572 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

# まず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