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