AtCoder が提供しているABC(AtCoder Beginner Contest)379 B問題をC++とPythonで解いてみました。ABC379は、2024年11月9日21:00に実施されました。
AtCoder の紹介はこちらに、プログラミングの方針はこちらに記事があります。
B問題 Strawberries(Difficulty : 39)
問題の詳細は、リンク先をご覧ください。
指定された文字が繰り返されるパターンを探します。AtCoder Problems による Difficulty は 39 でした。
解答案
C++ プログラム例(ABC379B)
文字 ‘O’ が $K$ 回連続するパターンを探します(14ー22行目)。
以下が、C++プログラムです。
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n, k;
cin >> n >> k;
string s;
cin >> s;
int result = 0;
int o_count = 0;
for (int i = 0; i < (int)s.length(); ++i) {
if (s[i] == 'O') {
++o_count;
if (o_count >= k) {
++result;
o_count = 0;
}
} else {
o_count = 0;
}
}
cout << result << endl;
return 0;
}
AC(Accepted=正しいプログラム)と判定されました。
Python プログラム例(ABC379B)
Python版も基本的な考え方は同じです。以下がプログラムです。
"""AtCoder Beginner Contest 379 B"""
n, k = map(int, input().split())
s = input()
result = 0
o_count = 0
for _, ch in enumerate(s):
if ch == 'O':
o_count += 1
if o_count >= k:
result += 1
o_count = 0
else:
o_count = 0
print(result)
こちらも「AC」と判定されました。
最後に
文字列クラスのメソッドや標準関数を使っても解けますが、自分で実装しました。
引き続き ABC の問題を紹介していきます。