第8回日本情報オリンピック 予選(過去問)から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
「競技プログラミング」を略して、「競プロ」などと呼ばれています。
#C - 連鎖
僕が作成、提出したコードは、以下のとおりです。
/* AtCoder Problems in C++ #C - 連鎖 https://atcoder.jp/contests/joi2009yo/tasks/joi2009yo_c 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; int main(void){ int n; cin>>n; vector<int> s(n); for(int i=0;i<n;i++){ cin>>s[i]; } int ans=n; for(int i=0;i<n;i++){ for(int j=1;j<=3;j++){ if(s[i]==j){ continue; } swap(j,s[i]); stack<pair<int,int>> st; int num=n; for(int k=0;k<n;k++){ if(st.empty()){ st.push({s[k],1}); }else if(st.top().first==s[k]){ st.top().second++; }else{ if(st.top().second>=4){ num-=st.top().second; st.pop(); } if(st.empty()){ st.push({s[k],1}); }else if(st.top().first==s[k]){ st.top().second++; }else{ st.push({s[k],1}); } } } if(st.top().second>=4){ num-=st.top().second; st.pop(); } ans=min(ans,num); swap(j,s[i]); } } cout<<ans<<endl; return 0; }