結果

問題 No.1150 シュークリームゲーム（Easy）
ユーザー たたき@競プロ
提出日時 2023-09-14 19:21:21
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 44 ms / 2,000 ms
コード長 905 bytes
コンパイル時間 5,971 ms
コンパイル使用メモリ 311,376 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-02 02:20:40
合計ジャッジ時間 8,885 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 43
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In lambda function:
main.cpp:27:3: warning: control reaches end of non-void function [-Wreturn-type]
   27 |   };
      |   ^

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint=modint998244353; //1000000007;
using ll=long long;
using pp=pair<int,int>;
#define sr string 
#define vc vector
#define fi first
#define se second
#define rep(i,n) for(int i=0;i<(int)n;i++)
#define pb push_back
#define all(v) v.begin(),v.end()
#define pque priority_queue
#define bpc(a) __builtin_popcount(a)
int main(){
  int n,s,t;cin>>n>>s>>t;
  s--;t--;
  vc<ll>v(n); rep(i,n)cin>>v[i];
  auto f=[&](int a)->int{
    int ds=abs(s-a); ds=min(ds,n-ds);
    int dt=abs(t-a); dt=min(dt,n-dt);
    if(ds>dt)return -1;
    if(ds<dt)return 1;
    if(ds==dt)return 0;
  };
  ll x=0; vc<ll>vx;
  rep(i,n)x+=v[i]*f(i);
  rep(i,n)if(!f(i))vx.pb(v[i]);
  sort(all(vx)); reverse(all(vx));
  //cout<<x<<"\n";
  rep(i,vx.size()){
    //cout<<vx[i]<<"\n";
    if(i==0)x+=vx[i];
    else x-=vx[i];
  }
  cout<<x;
}
0