JOI 2019/2020 本選 過去問から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
「競技プログラミング」を略して、「競プロ」などと呼ばれています。
#B - JJOOII 2 (JJOOII 2)
僕が作成、提出したコードは、以下のとおりです。
/* AtCoder Problems in C++ #B - JJOOII 2 (JJOOII 2) https://atcoder.jp/contests/joi2020ho/tasks/joi2020ho_b 提出コードの解答例 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; string s; cin>>s; vector<ll> a(n+1),b(n+1),c(n+1); for(ll i=0;i<n;i++){ a[i+1]=a[i]+(s[i]=='J'); b[i+1]=b[i]+(s[i]=='O'); c[i+1]=c[i]+(s[i]=='I'); } ll ia=0,ib=0,ic=0; const ll inf=1e+9; ll ans=inf; for(ll i=0;i<n;i++){ for(;ia<=n;ia++){ if(a[ia]-a[i]>=k){ break; } } if(ia>n){ break; } for(;ib<=n;ib++){ if(b[ib]-b[ia]>=k){ break; } } if(ib>n){ break; } for(;ic<=n;ic++){ if(c[ic]-c[ib]>=k){ break; } } if(ic>n){ break; } ans=min(ans,ic-i); } if(ans>=inf){ cout<<-1<<endl; }else{ cout<<ans-3*k<<endl; } return 0; }