JOI 予選 2008 B - JOIとIOI (AOJ 0522) (7Q, 難易度 2)

近年の ABC A 問題ではよくある「文字列の部分文字列を調べる」系の問題

• 問題へのリンク

問題概要

文字列 が与えられる。この文字列に含まれる "JOI" の個数と、"IOI" の個数をそれぞれ求めよ。

制約

• 

考えたこと

この問題よりも易しいバージョンとして、「文字列 中に含まれる文字 j の個数を求めよ」が考えられる。この問題は、次のような for 文で解ける。

int res = 0;
for (int i = 0; i < S.size(); i++) {
    if (S[i] == 'j') res++;
}

今回は、 に含まれるかを判定するものが「単一の文字」ではなく、「3 文字の文字列」である点が難しい。

しかし、実は、やるべきことはほとんど変わらない。文字列 の 文字目から数えて 3 文字分を取り出した文字列は、C++ では

S.substr(i, 3)

によって取得できる。これが "JOI" や "IOI" に一致するかどうかを調べていけばよい！

コード

for 文のループ変数 の範囲について考える。

「文字列 の 文字目から数えて 3 文字分」の末尾の文字は、 の 文字目であることに注意しよう。よって、下のコードでは、for 文の終端条件を i + 2 < S.size() とした。

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

int main() {
    string S;
    cin >> S;

    int joi = 0, ioi = 0;
    for (int i = 0; i + 2 < S.size(); i++) {
        if (S.substr(i, 3) == "JOI") joi++;
        if (S.substr(i, 3) == "IOI") ioi++;
    }
    cout << joi << endl;
    cout << ioi << endl;
}