寝癖頭の解法

勉強中の覚え書きを投稿、更新していきます。

AtCoder Problems in C++ #C - 連鎖

第8回日本情報オリンピック 予選(過去問)から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
競技プログラミング」を略して、「競プロ」などと呼ばれています。

#C - 連鎖

atcoder.jp

僕が作成、提出したコードは、以下のとおりです。

/*
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;
}