AtCoder Beginner Contestの過去問から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
「競技プログラミング」を略して、「競プロ」などと呼ばれています。
#E - Amusement Park
僕が作成、提出したコードは、以下のとおりです。
/* AtCoder Problems in C++ #E - Amusement Park https://atcoder.jp/contests/abc216/tasks/abc216_e 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; const ll N=1e5+5; ll n,k,a[N],ans; int main(void){ cin>>n>>k; for(ll i=1;i<=n;i++){ cin>>a[i]; } sort(a+1,a+n+1,[](ll i,ll j){return i>j;}); for(ll i=1;i<=n && k;i++){ ll c=a[i+1]; if(c==a[i]){ continue; } if((a[i]-c)*i<=k){ ans+=(a[i]-c)*(c+a[i]+1)/2*i; k-=(a[i]-c)*i; }else{ ll j=a[i]; for(;j>=c && k>=i;j--){ k-=i; ans+=j*i; } ans+=j*k; k=0; } } cout<<ans<<endl; return 0; }
AtCoder Beginner Contestは、オンラインジャッジによるプログラミングコンテストです。
日本語と英語に対応していて、週末ごとに実施されているみたいです。
https://practice.contest.atcoder.jp/tutorial
アカウントを登録すれば、誰でも参加できます。