AtCoder が提供しているABC(AtCoder Beginner Contest)392 B問題をC++とPythonで解いてみました。ABC392は、2025年2月8日21:00に実施されました。
AtCoder の紹介はこちらに、プログラミングの方針はこちらに記事があります。
B問題 Who is Missing?(Difficulty : 29)
問題の詳細は、リンク先をご覧ください。
出現した要素を管理するコンテナを用意します。AtCoder Problems による Difficulty は 29 でした。
解答案
C++ プログラム例(ABC392B)
配列 result
は、出現していない数値を真とする bool 配列です。この配列を更新して、最後に結果を出力します。
以下が、C++プログラムです。
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n, m;
cin >> n >> m;
vector<bool> result(n + 1, true);
result[0] = false;
for (int i = 0; i < m; ++i) {
int a;
cin >> a;
result[a] = false;
}
cout << n - m << endl;
for (int i = 1; i <= n; ++i) {
if (result[i]) {
cout << i << " ";
}
}
cout << endl;
return 0;
}
管理するコンテナを set にしました。最初に1からNまでの整数をすべて格納して、出現したら削除します。最後に set コンテナ result
のサイズと要素をすべて出力します。
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n, m;
cin >> n >> m;
set<int> result;
for (int i = 1; i <= n; ++i) {
result.insert(i);
}
for (int i = 0; i < m; ++i) {
int a;
cin >> a;
result.erase(a);
}
cout << result.size() << endl;
for (auto e : result) {
cout << e << " ";
}
cout << endl;
return 0;
}
どちらも AC(Accepted=正しいプログラム)と判定されました。
Python プログラム例(ABC392B)
Python版も基本的な考え方は同じです。以下がプログラムです。
"""AtCoder Beginner Contest 392 B"""
n, m = map(int, input().split())
a = list(map(int, input().split()))
result = [True] * (n + 1)
result[0] = False
for e in a:
result[e] = False
print(n - m)
for i in range(1, n + 1):
if result[i]:
print(i, end=" ")
print()
こちらも「AC」と判定されました。
最後に
このB問題も初心者に取り組みやすい難易度でした。
引き続き ABC の問題を紹介していきます。