結果

問題 No.972 選び方のスコア
ユーザー persimmon-persimmonpersimmon-persimmon
提出日時 2021-06-30 09:28:34
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 196 ms / 2,000 ms
コード長 2,287 bytes
コンパイル時間 152 ms
コンパイル使用メモリ 82,456 KB
実行使用メモリ 121,856 KB
最終ジャッジ日時 2024-06-26 20:25:26
合計ジャッジ時間 6,989 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 196 ms
114,384 KB
testcase_01 AC 188 ms
113,868 KB
testcase_02 AC 180 ms
107,392 KB
testcase_03 AC 113 ms
96,896 KB
testcase_04 AC 178 ms
108,416 KB
testcase_05 AC 175 ms
108,288 KB
testcase_06 AC 177 ms
108,544 KB
testcase_07 AC 154 ms
112,192 KB
testcase_08 AC 155 ms
112,476 KB
testcase_09 AC 142 ms
107,244 KB
testcase_10 AC 131 ms
112,864 KB
testcase_11 AC 132 ms
112,328 KB
testcase_12 AC 152 ms
113,272 KB
testcase_13 AC 140 ms
112,504 KB
testcase_14 AC 150 ms
113,272 KB
testcase_15 AC 170 ms
114,332 KB
testcase_16 AC 166 ms
115,036 KB
testcase_17 AC 165 ms
114,204 KB
testcase_18 AC 37 ms
51,712 KB
testcase_19 AC 189 ms
121,856 KB
testcase_20 AC 181 ms
107,648 KB
testcase_21 AC 188 ms
121,696 KB
testcase_22 AC 176 ms
108,288 KB
testcase_23 AC 189 ms
121,728 KB
testcase_24 AC 186 ms
108,032 KB
testcase_25 AC 37 ms
51,968 KB
testcase_26 AC 38 ms
52,388 KB
testcase_27 AC 38 ms
51,968 KB
testcase_28 AC 38 ms
51,968 KB
testcase_29 AC 38 ms
52,352 KB
testcase_30 AC 38 ms
52,224 KB
testcase_31 AC 37 ms
52,096 KB
testcase_32 AC 38 ms
52,480 KB
testcase_33 AC 41 ms
58,112 KB
testcase_34 AC 38 ms
52,096 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

"""
選択する数字は奇数個でよいか。
奇数個取る場合、中央値はaのいずれかになるので扱いやすい
偶数個取る場合、中央値の候補がn^2個あり扱いが難しい。

偶数個取るとする。
bの中央値をmとし、mを作る二つの要素をm-d,m+dとする。簡単のため要素はすべて異なるとする。
sum0=sum([x-m for x in b if x<=m-d])
sum1=sum([x-m for x in b if x>=m+d])
n0=len([x-m for x in b if x<=m-d])
n1=len([x-m for x in b if x>=m+d])
sum0+sum1がスコア

このときbから要素m+dを取り除くと中央値はm-dになる。
新しいスコアは
new_sum0=sum([x-(m-d) for x in b if x<=m-d])=sum([x-m for x in b if x<=m-d]) + d * n0 = sum0 + d * n0
new_sum1=sum([x-(m-d) for x in b if x>m+d])=sum([x-(m-d) for x in b if x>=m+d]) - (m+d-(m-d))
=sum([x-m for x in b if x>=m+d]) - 2*d + n1*d = sum1 + d * (n1-2)
->sum0+sum1+d*(n0+n1-2)
n0==n1,n0>0なのでn0+n1>=2.偶数個取った場合、真ん中の大きい方を取り除いた数列にしても損することはない。
よって奇数個取ることだけを考えてよい。
"""
def main0(n,a):
  # O(N^2)
  ans=0
  a.sort()
  for i in range(n):
    # a[i]を中央値とする。
    now=0
    # 貪欲に取っていく。要素を増やしたとき負が加算されるまで取り続ける。
    for j in range(n):
      if -n<=-i-j and 0<=i-1-j and (a[-1-j]-a[i]) + (a[i-1-j]-a[i])>=0:
        now+=a[-1-j]-a[i]
        now+=a[i-1-j]-a[i]
      else:
        break
    ans=max(ans,now)
  return ans

def main1(n,a):
  # main0の貪欲部分を二分探索する。
  ans=0
  a.sort()
  sa=[0]
  for x in a:sa.append(sa[-1]+x)
  for i in range(n):
    # a[i]を中央値とする。
    # 貪欲に取っていく。二分探索。2x+1個の要素からなる数列。xを求める。x=0なら0。x>=1の範囲を探索
    l,r=1,min(i,n-i-1)+1
    while r-l>1:
      x=(r+l)//2
      if (a[-x]-a[i])+(a[i-x]-a[i])>=0:
        l,r=x,r
      else:
        l,r=l,x
    # x=l+1
    x=l
    #print(i,x,sa)
    tmp=sa[n]-sa[n-x] + sa[i]-sa[i-x]
    tmp-=a[i]*(2*x)
    ans=max(ans,tmp)
  return ans

if __name__=='__main__':
  n=int(input())
  a=list(map(int,input().split()))
  ret1=main1(n,a)
  print(ret1)
  #ret0=main0(n,a)
  #print(ret0)
0