結果

問題 No.198 キャンディー・ボックス２
ユーザー あかりき
提出日時 2021-03-15 00:07:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 173 ms / 1,000 ms
コード長 1,146 bytes
コンパイル時間 945 ms
コンパイル使用メモリ 82,600 KB
実行使用メモリ 101,032 KB
最終ジャッジ日時 2024-11-06 12:12:51
合計ジャッジ時間 5,739 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

b=int(input())
n=int(input())
c=[int(input()) for i in range(n)]
c.sort()
ans=[]
if n//2==1:
  t=c[n//2]
  for i in range(max(t-10**5,0),t+10**5):
    ct=0
    y=b
    for j in range(n):
      if c[j]>=i:
        ct+=c[j]-i
        y+=c[j]-i
    for j in range(n):
      if c[j]<i:
        ct+=i-c[j]
        y-=i-c[j]
    if y>=0:
      ans.append(ct)
else:
  t1=c[n//2-1]
  t2=c[n//2]
  for i in range(max(t1-10**5,0),t1+10**5):
    ct=0
    y=b
    for j in range(n):
      if c[j]>=i:
        ct+=c[j]-i
        y+=c[j]-i
    for j in range(n):
      if c[j]<i:
        ct+=i-c[j]
        y-=i-c[j]
    if y>=0:
      ans.append(ct)
  for i in range(max(t2-10**5,0),t2+10**5):
    ct=0
    y=b
    for j in range(n):
      if c[j]>=i:
        ct+=c[j]-i
        y+=c[j]-i
    for j in range(n):
      if c[j]<i:
        ct+=i-c[j]
        y-=i-c[j]
    if y>=0:
      ans.append(ct)

x=sum(c)+b
t3=x//n
for i in range(max(t3-10**5,0),t3+10**5):
  ct=0
  y=b
  for j in range(n):
    if c[j]>=i:
      ct+=c[j]-i
      y+=c[j]-i
  for j in range(n):
    if c[j]<i:
      ct+=i-c[j]
      y-=i-c[j]
  if y>=0:
    ans.append(ct)
print(min(ans))
0