AtCoder

ABC388 B問題(Heavy Snake)を解く

AtCoder_ABC388_B

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

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

B問題 Heavy Snake(Difficulty : 30)

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

ABC388 B問題 Heavy Snake

問題の指示に従い計算するだけです。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 の問題を紹介していきます。

COMMENT

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

CAPTCHA