AtCoder が提供しているABC(AtCoder Beginner Contest)388 B問題をC++とPythonで解いてみました。ABC388は、2025年1月11日21:00に実施されました。
AtCoder の紹介はこちらに、プログラミングの方針はこちらに記事があります。
B問題 Heavy Snake(Difficulty : 30)
問題の詳細は、リンク先をご覧ください。
問題の指示に従い計算するだけです。AtCoder Problems による Difficulty は 30 でした。
解答案
C++ プログラム例(ABC388B)
変数 $k$ の値を $1$ から $D$ まで変化させ、各値に対してヘビの重さの最大値を求めます。C++ の変数名は通常小文字で記述しますが、変数名 ‘l’(小文字のエル)は数字の ‘1’ と紛らわしいため、大文字の ‘L’ を使用しています。
以下が、C++プログラムです。
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n, d;
cin >> n >> d;
vector<int> t(n);
vector<int> L(n);
for (int i = 0; i < n; ++i) {
cin >> t[i] >> L[i];
}
for (int k = 1; k <= d; ++k) {
int result = 0;
for (int i = 0; i < n; ++i) {
result = max(result, t[i] * (L[i] + k));
}
cout << result << endl;
}
return 0;
}
AC(Accepted=正しいプログラム)と判定されました。
Python プログラム例(ABC388B)
Python版も基本的な考え方は同じです。以下がプログラムです。
"""AtCoder Beginner Contest 388 B"""
n, d = map(int, input().split())
t = [0] * n
L = [0] * n
for i in range(n):
t[i], L[i] = map(int, input().split())
for k in range(1, d + 1):
result = 0
for i in range(n):
result = max(result, t[i] * (L[i] + k))
print(result)
こちらも「AC」と判定されました。
最後に
この問題は数学的なアプローチで解くことも可能かもしれませんが、計算量的に直接計算しても十分間に合うため、その方法を採用しました。公式解説でも同様の方針が採用されています。
引き続き ABC の問題を紹介していきます。