結果

問題 No.198 キャンディー・ボックス2
ユーザー aka_satana_haaka_satana_ha
提出日時 2017-11-28 15:10:43
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 33 ms / 1,000 ms
コード長 719 bytes
コンパイル時間 353 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 10,752 KB
最終ジャッジ日時 2024-11-16 10:14:28
合計ジャッジ時間 2,451 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

B = int(input())
N = int(input())
C=[]
for i in range(N):
    C.append(int(input()))

C.sort()
C.reverse()
C_sum=sum(C)
ave_max=int((C_sum+B)/N)#最大の個数
ave_min=0

#三分探索
tmp_min=0xFFFFFFFF
while(1):
    x0=ave_min+int((ave_max-ave_min)/3)
    x1=ave_min+int((ave_max-ave_min)/3*2)
    # print(x0,x1)
    if abs(x0-x1)<2:
        break
    x0_sum=0
    x1_sum=0
    for i in range(N):
        x0_sum+=abs(C[i]-x0)
        x1_sum+=abs(C[i]-x1)
    if x0_sum>x1_sum:
        ave_min=x0
    else:
        ave_max=x1

sum_min=float('inf')
for i in range(ave_min,ave_max+1):
    tmp_sum=0
    for j in range(N):
        tmp_sum+=abs(C[j]-i)
    if tmp_sum<sum_min:
        sum_min=tmp_sum

print(int(sum_min))
0