AtCoder が提供しているABC（AtCoder Beginner Contest）356 B問題をC++とPythonで解いてみました。ABC356は、2024年6月1日21:00に実施されました。
AtCoder の紹介はこちらに、プログラミングの方針はこちらに記事があります。
B問題 Nutrients（Difficulty : 40）
問題の詳細は、リンク先をご覧ください。
与えられた行列の各列の総和を求めます。AtCoder Problems による Difficulty は 40 でした。
解答案
C++ プログラム例（ABC356B）
先に数列 A を読み込みます。
次に行列 X を読み込み、読みながら各列の和 sum_x を更新します（12ー19行目）。
数列 A と各列の和 sum_x を比較して、A の方が大きい要素があれば、この栄養素について、食事によって十分な量を摂取できていないことになります。この場合は、”No” を出力します。このような要素がなければ、”Yes” を出力します。
以下が、C++プログラムです。
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n, m;
cin >> n >> m;
vector<int> a(m);
for (int i = 0; i < m; ++i) {
cin >> a[i];
}
vector<int> sum_x(m, 0);
for (int i = 0; i < n; ++i) {
for (int j = 0; j < m; ++j) {
int x;
cin >> x;
sum_x[j] += x;
}
}
bool result = true;
for (int i = 0; i < m; ++i) {
if (a[i] > sum_x[i]) {
result = false;
}
}
if (result) {
cout << "Yes" << endl;
} else {
cout << "No" << endl;
}
return 0;
}AC（Accepted＝正しいプログラム）と判定されました。
Python プログラム例（ABC356B）
Python版も基本的な考え方は同じです。以下がプログラムです。
"""AtCoder Beginner Contest 356 B"""
n, m = map(int, input().split())
a = list(map(int, input().split()))
sum_x = [0] * m
for i in range(n):
x = list(map(int, input().split()))
for j in range(m):
sum_x[j] += x[j]
result = True
for i in range(m):
if a[i] > sum_x[i]:
result = False
print("Yes" if result else "No")こちらも「AC」と判定されました。
最後に
2次元の配列が問題の入力として与えられます。ただし、各列の和を求めることにより、1次元の配列で処理できました。
引き続き ABC の問題を紹介していきます。
