結果

問題 No.31 悪のミックスジュース
ユーザー szkhts
提出日時 2023-02-12 08:19:59
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 744 ms / 5,000 ms
コード長 767 bytes
コンパイル時間 246 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,520 KB
最終ジャッジ日時 2024-07-16 02:14:44
合計ジャッジ時間 8,764 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

n, v = map(int, input().split())
c = list(map(int, input().split()))
for i in range(1, n):
    c[i] += c[i - 1]

v -= n
if v <= 0:
    print(c[-1])
    exit(0)

num_cost_pairs = list(enumerate(c, 1))

class Rational:
    def __init__(self, p):
        # p[1] / p[0]
        self.n = p[1]
        self.d = p[0]
    def __lt__(self, r):
        return self.n * r.d < self.d * r.n

num_cost_pairs.sort(key=Rational)

(mn, mc), *others = num_cost_pairs
INF = 10 ** 18
MX = 100 * 100
dp = [INF] * (MX + 1)
dp[0] = 0
for i in range(MX):
    for num, cost in others:
        j = i + num
        if j <= MX:
            dp[j] = min(dp[j], dp[i] + cost)
ans = INF
for vi in range(v % mn, min(MX, v) + 1, mn):
    ans = min(ans, dp[vi] + (v - vi) // mn * mc)
print(ans + c[-1])
0