dwangoプログラミングコンテスト予選 B問題 - ニコニコ文字列

Source

dwangoプログラミングコンテスト予選
問題文

問題概要

長さ $N$ の数字のみからなる文字列 $S$ が与えられる.
ニコニコ文字列とは $\verb|25|$,$\verb|2525|$,$\verb|252525|$,$\ldots$ のように,$\verb|25|$ を $1$ つ以上連結した文字列です.
$S$ の中に部分文字列として,ニコニコ文字列が何箇所に登場するかを求める問題.

解法

$\verb|25|$ が連続している極大の部分を見つけてきて,各部分に対して,ニコニコ文字列の数を数えて足し上げる.
$\verb|25|$ が $k$ 回繋がっているなら,そこの部分に含まれるニコニコ文字列の数は,始点と終点の選び方のパターンの数で $(k+1)k/2$ 通り.

C++によるスパゲッティなソースコード

#include<bits/stdc++.h>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)

#define ll long long

int reader(char c[]){int i,s=0;for(;;){mygc(i);if(i!=' '&&i!='\n'&&i!='\r'&&i!='\t'&&i!=EOF) break;}c[s++]=i;for(;;){mygc(i);if(i==' '||i=='\n'||i=='\r'||i=='\t'||i==EOF) break;c[s++]=i;}c[s]='\0';return s;}
void writer(ll x, char c){int s=0,m=0;char f[20];if(x<0)m=1,x=-x;while(x)f[s++]=x%10,x/=10;if(!s)f[s++]=0;if(m)mypc('-');while(s--)mypc(f[s]+'0');mypc(c);}
template<class T> void writerLn(T x){writer(x,'\n');}

int N;
char in[1000000];

int main(){
  int i, j, k;
  ll res = 0, x;

  N = reader(in);
  rep(i,N){
    j = i;
    while(j+1<N && in[j]=='2' && in[j+1]=='5') j+=2;
    if(j==i) continue;

    x = (j-i) / 2;
    res += (x+1) * x / 2;

    i = j-1;
  }

  writerLn(res);

  return 0;
}

Current time: 2017年09月21日05時07分32秒
Last modified: 2015年01月28日23時21分37秒 (by laycrs)
Tags: Competitive_Programming AtCoder dwango2015-prelims
トップページに戻る

Logged in as: unknown user (not login)

ログイン: