AtCoder が提供しているABC(AtCoder Beginner Contest)385 A問題をC++とPythonで解いてみました。ABC385は、2024年12月21日21:00に実施されました。
AtCoder の紹介はこちらに、プログラミングの方針はこちらに記事があります。
A問題 Equally(Difficulty : 14)
問題の詳細は、リンク先をご覧ください。
与えられた整数を2組か3組に分けて確認します。AtCoder Problems による Difficulty は 14 でした。
解答案
C++ プログラム例(ABC385A)
3つの整数を読み込み、ソートします。ソート後、最小値と最大値が一致していれば、3つの値はすべて同じです。小さい2つの値の和が最大の値と一致していれば、条件を満たします。
以下が、C++プログラムです。
#include <bits/stdc++.h>
using namespace std;
int main()
{
vector<int> a(3);
for (int i = 0; i < 3; ++i) {
cin >> a[i];
}
sort(a.begin(), a.end());
bool result = false;
if (a[0] == a[2]) {
result = true;
}
if (a[0] + a[1] == a[2]) {
result = true;
}
if (result) {
cout << "Yes" << endl;
} else {
cout << "No" << endl;
}
return 0;
}
AC(Accepted=正しいプログラム)と判定されました。
Python プログラム例(ABC385A)
Python版も基本的な考え方は同じです。以下がプログラムです。
"""AtCoder Beginner Contest 385 A"""
a = list(map(int, input().split()))
a.sort()
result = False
if a[0] == a[2]:
result = True
if a[0] + a[1] == a[2]:
result = True
print("Yes" if result else "No")
こちらも「AC」と判定されました。
最後に
この問題では、入力例2がなければ、3つの値がすべて同じ場合に条件を満たすことに気付かない可能性がありました。難易度が低い問題でも、すべての入力例を確認することが重要だと考えています。
引き続き ABC の問題を紹介していきます。