AtCoder が提供しているABC(AtCoder Beginner Contest)361 A問題をC++とPythonで解いてみました。ABC361は、2024年7月6日21:00に実施されました。
AtCoder の紹介はこちらに、プログラミングの方針はこちらに記事があります。
A問題 Insert(Difficulty : 15)
問題の詳細は、リンク先をご覧ください。
配列に要素を挿入するだけです。AtCoder Problems による Difficulty は 15 でした。
解答案
C++ プログラム例(ABC361A)
出力するベクトルコンテナ b に、読み込んだ a の要素をコピーしていきます(15行目)。k 番目の要素の後に x を push_back します(16ー17行目)。
以下が、C++プログラムです。
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n, k, x;
cin >> n >> k >> x;
vector<int> a(n);
for (int i = 0; i < n; ++i) {
cin >> a[i];
}
vector<int> b;
for (int i = 0; i < n; ++i) {
b.push_back(a[i]);
if (i + 1 == k) {
b.push_back(x);
}
}
for (int i = 0; i <= n; ++i) {
cout << b[i] << " \n"[i == n];
}
return 0;
}
AC(Accepted=正しいプログラム)と判定されました。
Python プログラム例(ABC361A)
Python版も基本的な考え方は同じです。以下がプログラムです。
"""AtCoder Beginner Contest 361 A"""
n, k, x = map(int, input().split())
a = list(map(int, input().split()))
b = []
for i in range(n):
b.append(a[i])
if i + 1 == k:
b.append(x)
print(*b)
リストのinsertメソッドを使うと以下のように簡単に書けます。
"""AtCoder Beginner Contest 361 A"""
n, k, x = map(int, input().split())
a = list(map(int, input().split()))
a.insert(k, x)
print(*a)
どちらも「AC」と判定されました。
最後に
今回は、A問題正解者が10409人でした。前回のABC360は8406人で、前々回のABC359は10583人でした。事情は分かりませんが、前回は参加者が少なかったようです。
引き続き ABC の問題を紹介していきます。