結果
問題 | No.31 悪のミックスジュース |
ユーザー | rpy3cpp |
提出日時 | 2015-09-04 13:55:16 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 147 ms / 5,000 ms |
コード長 | 1,144 bytes |
コンパイル時間 | 69 ms |
コンパイル使用メモリ | 10,908 KB |
実行使用メモリ | 8,780 KB |
最終ジャッジ日時 | 2023-10-12 17:18:51 |
合計ジャッジ時間 | 1,583 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 17 ms
8,256 KB |
testcase_01 | AC | 90 ms
8,632 KB |
testcase_02 | AC | 103 ms
8,700 KB |
testcase_03 | AC | 132 ms
8,696 KB |
testcase_04 | AC | 16 ms
8,360 KB |
testcase_05 | AC | 16 ms
8,352 KB |
testcase_06 | AC | 16 ms
8,212 KB |
testcase_07 | AC | 25 ms
8,248 KB |
testcase_08 | AC | 23 ms
8,200 KB |
testcase_09 | AC | 147 ms
8,780 KB |
testcase_10 | AC | 16 ms
8,248 KB |
testcase_11 | AC | 18 ms
8,364 KB |
testcase_12 | AC | 62 ms
8,460 KB |
testcase_13 | AC | 16 ms
8,212 KB |
testcase_14 | AC | 20 ms
8,328 KB |
testcase_15 | AC | 16 ms
8,244 KB |
testcase_16 | AC | 17 ms
8,244 KB |
ソースコード
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 record = None for k, d in enumerate(Ds, 1): det = k * cost - n * d if det > 0: n = k cost = d record = k - 1 return record 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)) eff = select_most_efficient(Ds) return calc_cost(N, V, Cs, Ds, eff) + sum(Cs) def calc_cost(N, V, Cs, Ds, m): d = Ds[m] dp = get_dp(N, Ds, m) # V - N * m <= i * (m + 1) <= V ibegin = max(0, 1 + (V - N * m - 1) // (m + 1)) iend = 1 + V // (m + 1) return min(d * i + dp[V - (m + 1) * i] for i in range(ibegin, iend)) def get_dp(N, Ds, m): dp = [0] + [float('inf')] * (N * m) for i in range(1, N * m + 1): dp[i] = min(dp[i - j] + d for j, d in enumerate(Ds[:i], 1)) return dp if __name__ == '__main__': N, V, Cs = read_data() print(solve(N, V, Cs))