結果

問題 No.31 悪のミックスジュース
コンテスト
ユーザー te-sh
提出日時 2017-02-15 15:55:48
言語 D
(dmd 2.109.1)
結果
WA  
実行時間 -
コード長 1,068 bytes
コンパイル時間 1,113 ms
コンパイル使用メモリ 120,720 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-12 07:03:20
合計ジャッジ時間 2,639 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 2 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.algorithm, std.conv, std.range, std.stdio, std.string;
import std.typecons;  // Tuple, Nullable, BigFlags

void main()
{
  auto rd = readln.split, n = rd[0].to!size_t, v = rd[1].to!long;
  auto ci = readln.split.to!(long[]);

  foreach (i, c; ci.drop(1)) ci[i + 1] = ci[i] + c;
  v -= n;

  if (v <= 0) {
    writeln(ci.back);
    return;
  }

  auto k = ci.enumerate!long
    .map!(ec => Tuple!(long, real)(ec.index, ec.value.to!real / (ec.index + 1))).array
    .sort!"a[1] < b[1]".front[0];

  writeln(k);

  auto dp = new long[]((k + 1) * n);
  dp[] = long.max;
  dp[0] = 0;

  foreach (i; n.iota) {
    auto dp2 = dp.dup;
    foreach (dpi, dpv; dp) {
      if (dpv != long.max) {
        foreach (j; (k + 1).iota) {
          auto idx = dpi + (i + 1) * j;
          if (idx < dp.length)
            dp2[idx] = min(dp2[idx], dpv + ci[i] * j);
        }
      }
    }
    dp = dp2;
  }

  auto r = dp.enumerate!long
    .filter!(d => d.value != long.max)
    .map!(d => d.value + (v - d.index + k) / (k + 1) * ci[k]);

  writeln(r.fold!min + ci.back);
}
0