Submission #1675712

---

Source Code Expand

#include "bits/stdc++.h"
using namespace std;
     
#define DEBUG(x) cout<<#x<<": "<<x<<endl;
#define DEBUG_VEC(v) cout<<#v<<":";for(int i=0;i<v.size();i++) cout<<" "<<v[i]; cout<<endl
     
typedef long long ll;
#define vi vector<int>
#define vl vector<ll>
#define vii vector< vector<int> >
#define vll vector< vector<ll> >
#define vs vector<string>
#define pii pair<int,int>
#define pis pair<int,string>
#define psi pair<string,int>
const int inf = 1000000001;
const ll INF = 1e16;
#define MOD 1000000007
#define mod 1000000009
#define pi 3.14159265358979323846
#define Sp(p) cout<<setprecision(15)<<fixed<<p<<endl;
int dx[4] = { 1,0,-1,0 }, dy[4] = { 0,1,0,-1 };
int dx2[8] = { 1,1,0,-1,-1,-1,0,1 }, dy2[8] = { 0,1,1,1,0,-1,-1,-1 };

int main(){
  string s;
  cin>>s;
  int i, j, n=s.size();
  ll sum=0;
  char pre='0';
  vl a;
  int cnt=0;
  for(i=0;i<n;i++){
    if(s[i]!='2' && s[i]!='5'){
      if(cnt){
	a.push_back(cnt);
	cnt=0;
      }
    }
    else if(s[i]=='2'){
      if(pre != '5'){
	if(cnt){
	  a.push_back(cnt);
	  cnt=0;
	}
      }
    }
    else{
      if(pre == '2'){
	cnt++;
      }else{
	if(cnt){
	  a.push_back(cnt);
	  cnt=0;
	}
      }
    }
    pre = s[i];
  }
  if(cnt){
    a.push_back(cnt);
  }
  for(i=0;i<a.size();i++){
    sum += a[i]*(a[i] + 1) / 2;
  }
  cout<<sum<<endl;
}

Submission Info

Judge Result