AtCoder

ABC392 B問題(Who is Missing?)を解く

AtCoder_ABC392_B

AtCoder が提供しているABC(AtCoder Beginner Contest)392 B問題をC++とPythonで解いてみました。ABC392は、2025年2月8日21:00に実施されました。

AtCoder の紹介はこちらに、プログラミングの方針はこちらに記事があります。

B問題 Who is Missing?(Difficulty : 29)

問題の詳細は、リンク先をご覧ください。

ABC392 B問題 Who is Missing?

出現した要素を管理するコンテナを用意します。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 の問題を紹介していきます。

COMMENT

メールアドレスが公開されることはありません。 が付いている欄は必須項目です

CAPTCHA