結果

問題 No.31 悪のミックスジュース
ユーザー rpy3cpprpy3cpp
提出日時 2015-09-02 08:07:37
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 1,462 bytes
コンパイル時間 276 ms
コンパイル使用メモリ 11,004 KB
実行使用メモリ 8,452 KB
最終ジャッジ日時 2023-09-25 21:53:38
合計ジャッジ時間 1,626 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge13
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 AC 16 ms
8,452 KB
testcase_03 AC 16 ms
8,388 KB
testcase_04 AC 24 ms
8,384 KB
testcase_05 AC 23 ms
8,376 KB
testcase_06 AC 17 ms
8,396 KB
testcase_07 AC 16 ms
8,304 KB
testcase_08 AC 17 ms
8,440 KB
testcase_09 WA -
testcase_10 AC 17 ms
8,412 KB
testcase_11 WA -
testcase_12 AC 16 ms
8,320 KB
testcase_13 AC 16 ms
8,308 KB
testcase_14 WA -
testcase_15 AC 16 ms
8,388 KB
testcase_16 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import itertools


def read_data():
    N, V = map(int, input().split())
    Cs = list(map(int, input().split()))
    return N, V, Cs


def select_most_efficient(Ds):
    n = 0
    cost = 1
    records = []
    for k, d in enumerate(Ds, 1):
        det = k * cost - n * d
        if det > 0:
            n = k
            cost = d
            records = [k - 1]
        elif det == 0:
            records.append(k - 1)
    return records


def solve(N, V, Cs):
    if V <= N:
        return sum(Cs)
    if N == 1:
        return Cs[0] * V
    V -= N
    Ds = list(itertools.accumulate(Cs))
    effs = select_most_efficient(Ds)
    min_cost = Cs[0] * V
    for idx in effs:
        cost = calc_cost(V, Cs, Ds, idx)
        if cost < min_cost:
            min_cost = cost
    return min_cost + sum(Cs)


def calc_cost(V, Cs, Ds, idx):
    m, r = divmod(V, idx + 1)
    cost = Ds[idx] * m
    if r == 0:
        return cost
    dp_head = get_dp(r, r, Cs[:r])
    if m == 0 or idx == len(Ds) - 1:
        return cost + dp_head[r]
    dp_tail = get_dp(r, m, Cs[idx + 1:])
    return cost + min(a + b for a, b in zip(dp_head[::-1], dp_tail))


def get_dp(r, m, Cs):
    Ds = itertools.accumulate(Cs)
    dp = [0] + [float('inf')] * r
    for i, d in enumerate(Ds, 1):
        if i > m or i > r:
            break
        for j in range(i, min(r, m * i) + 1, i):
            dp[j] = min(dp[j], dp[j - i] + d)
    return dp

N, V, Cs = read_data()
print(solve(N, V, Cs))
0