AtCoder Beginner Contestの過去問から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
「競技プログラミング」を略して、「競プロ」などと呼ばれています。
#B - Better Students Are Needed!
僕が作成、提出したコードは、以下のとおりです。
/* AtCoder Problems in C++ #B - Better Students Are Needed! https://atcoder.jp/contests/abc260/tasks/abc260_b 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; int main(void){ ll n,x,y,z; cin>>n>>x>>y>>z; vector<pair<ll,ll>> a,aa,b,bb,sum; vector<ll> num; for(ll i=0;i<n;i++){ ll A; cin>>A; a.push_back({A,-i}); aa.push_back({A,-i}); } sort(a.rbegin(),a.rend()); for(ll i=0;i<x;i++){ num.push_back(-a[i].second); } for(ll i=0;i<n;i++){ bool tf=true; ll B; cin>>B; bb.push_back({B,-i}); for(ll j=0;j<x;j++){ if(-a[j].second==i){ tf=false; } } if(tf){ b.push_back({B,-i}); } } sort(b.rbegin(),b.rend()); for(ll i=0;i<y;i++){ num.push_back(-b[i].second); } for(ll i=0;i<n;i++){ bool tf=true; for(ll j=0;j<x;j++){ if(-a[j].second==i){ tf=false; } } for(ll j=0;j<y;j++){ if(-b[j].second==i){ tf=false; } } if(tf){ sum.push_back({aa[i].first+bb[i].first,-i}); } } sort(sum.rbegin(),sum.rend()); for(ll i=0;i<z;i++){ num.push_back(-sum[i].second); } sort(num.begin(),num.end()); for(ll i=0;i<num.size();i++){ cout<<num[i]+1<<endl; } return 0; }
AtCoder Beginner Contestは、オンラインジャッジによるプログラミングコンテストです。
日本語と英語に対応していて、週末ごとに実施されているみたいです。
https://practice.contest.atcoder.jp/tutorial
アカウントを登録すれば、誰でも参加できます。