AtCoder が提供しているABC(AtCoder Beginner Contest)337 のB問題をC++とPythonで解いてみました。ABC337は、2024年1月20日21:00に実施されました。
AtCoder の紹介はこちらに、プログラミングの方針はこちらに記事があります。
B問題 Extended ABC(Difficulty : 97)
問題はリンク先をご覧ください。
いろいろな解き方がある問題です。AtCoder Problems による Difficulty は 97 でした。
解答案
公式解説にも記載がありますが、複数の解き方があります。この記事では、2つの解き方を紹介します。
C++ プログラム例(ABC337B)
正規表現を使ったプログラムを紹介します。
“^A*B*C*$” は、先頭から任意個の ‘A’、その後に任意個の ‘B’ が続き、その後に任意個の ‘C’ で終わる文字列を表します。任意個は0個を含みます。
以下が、C++プログラムとなります。
#include <bits/stdc++.h>
using namespace std;
int main()
{
string s;
cin >> s;
bool result = regex_match(s.begin(), s.end(), regex("^A*B*C*$"));
if (result) {
cout << "Yes" << endl;
} else {
cout << "No" << endl;
}
return 0;
}
別のプログラムを示します。
文字を先頭から走査して、文字コードが小さくなる箇所があれば、”No” となります。それ以外は、”Yes” とすれば良いです。以下がこの方針のプログラムです。
#include <bits/stdc++.h>
using namespace std;
int main()
{
string s;
cin >> s;
bool result = true;
for (int i = 0; i < s.length() - 1; ++i) {
if (s[i] > s[i + 1]) {
result = false;
}
}
if (result) {
cout << "Yes" << endl;
} else {
cout << "No" << endl;
}
return 0;
}
どちらも AC(Accepted=正しいプログラム)と判定されました。
Python プログラム例(ABC337B)
Python 版も基本的な考え方は同じです。正規表現版は以下となります。
"""AtCoder Beginner Contest 337 B"""
import re
s = input()
result = re.match("^A*B*C*$", s)
print("Yes" if result else "No")
文字コードが減少していないか確認する版は以下となります。
"""AtCoder Beginner Contest 337 B"""
s = input()
result = True
for i in range(len(s) - 1):
if s[i] > s[i + 1]:
result = False
print("Yes" if result else "No")
こちらも「AC」と判定されました。
最後に
今回はコンテストには参加できませんでしたが、問題を読んで後者の解法を思いつきました。正規表現は、今後のためになる可能性があると考えて、紹介しました。
引き続き ABC の問題を紹介していきます。