寝癖頭の解法

小学生の目線から、勉強中の覚え書きを投稿、更新していきます。

AtCoder Problems in C++ #1 - 電飾 (Illumination)

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

#1 - 電飾 (Illumination)

atcoder.jp

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

/*
AtCoder Problems in C++
#1 - 電飾 (Illumination)
https://atcoder.jp/contests/joi2013ho/tasks/joi2013ho1
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int main(void){
  int n;
  cin>>n;
  vector<int> a(n),b(n);
  for(int i=0;i<n;i++){
    cin>>a[i];
  }
  int c=0,d=0;
  for(int i=0;i<n;i++){
    if(d>0){
      if(a[i]==a[i-1]){
        b[c++]=d;
        d=0;
      }
    }
    d++;
  }
  if(d>0){
    b[c++]=d;
  }
  int e=0,ans=0;
  for(int i=0;i<c;i++){
    e+=b[i];
    if(i>2){
      e-=b[i-3];
    }
    if(ans<e){
      ans=e;
    }
  }
  cout<<ans<<endl;
  return 0;
}