結果

問題 No.972 選び方のスコア
ユーザー beetbeet
提出日時 2020-01-17 22:30:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 128 ms / 2,000 ms
コード長 1,072 bytes
コンパイル時間 2,183 ms
コンパイル使用メモリ 199,872 KB
最終ジャッジ日時 2025-01-08 18:58:33
ジャッジサーバーID
(参考情報)
judge2 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
using Int = long long;
template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}

//INSERT ABOVE HERE
signed main(){
  Int n;
  cin>>n;
  vector<Int> as(n);
  for(Int i=0;i<n;i++) cin>>as[i];
  sort(as.begin(),as.end());

  vector<Int> sm(n+1,0);
  for(Int i=0;i<n;i++) sm[i+1]=sm[i]+as[i];

  Int ans=0;
  // odd
  for(Int c=0;c<n;c++){
    Int l=0,r=min(c+1,n-c);
    while(l+1<r){
      Int m=(l+r)>>1;
      if(as[c]+as[c]<=as[c-m]+as[n-m]) l=m;
      else r=m;
    }
    // [c-l, c], [n-l, n)
    Int cnt=l*2+1;
    chmax(ans,sm[c+1]-sm[c-l]+sm[n]-sm[n-l]-as[c]*cnt);
  }
  // cout<<ans<<endl;

  // even
  for(Int c=0;c+1<n;c++){
    Int l=0,r=min(c+1,n-(c+1));
    while(l+1<r){
      Int m=(l+r)>>1;
      if(as[c]+as[c+1]<=as[c-m]+as[n-m]) l=m;
      else r=m;
    }
    // [c-l, c+1], [n-l, n)
    Int cnt=l*2+2;
    chmax(ans,sm[c+2]-sm[c-l]+sm[n]-sm[n-l]-(as[c]+as[c+1])*cnt/2);
  }

  cout<<ans<<endl;
  return 0;
}
0