結果

問題 No.162 8020運動
ユーザー te-sh
提出日時 2017-04-28 17:31:10
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 864 ms / 5,000 ms
コード長 1,197 bytes
コンパイル時間 2,595 ms
コンパイル使用メモリ 156,152 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-12 18:59:12
合計ジャッジ時間 17,640 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.algorithm, std.conv, std.range, std.stdio, std.string;
import std.math;      // math functions

void main()
{
  auto a = 80 - readln.chomp.to!int;
  auto rd = readln.split.map!(s => s.to!real / 100).array, p0 = rd[0], p1 = rd[1], p2 = rd[2];

  auto dp = new real[][](a + 1, 15);

  real prob(ulong b, int n)
  {
    real r = 1;
    r *= b.bitTest(0)   ? 1 - p1 : p1;
    r *= b.bitTest(n-1) ? 1 - p1 : p1;
    foreach (i; 1..n-1)
      r *= b.bitTest(i) ? 1 - p2 : p2;
    return r;
  }

  int[] conts(ulong b, int n)
  {
    auto r = new int[](0), c = 0;
    foreach (i; 0..n) {
      if (b.bitTest(i)) {
        ++c;
      } else if (c) {
        r ~= c; c = 0;
      }
    }
    if (c) r ~= c;
    return r;
  }

  real calc(int y, int n)
  {
    if (y == 0) return n;
    if (n == 1) return (1 - p0) * calc(y - 1, n);
    if (!dp[y][n].isNaN) return dp[y][n];

    real e = 0;
    foreach (i; 0..(1 << n)) {
      auto p = prob(i, n);
      auto ci = conts(i, n);
      e += p * ci.map!(c => calc(y - 1, c)).sum;
    }
    return dp[y][n] = e;
  }

  writefln("%.7f", calc(a, 14) * 2);
}

pragma(inline) {
  pure bool bitTest(T)(T n, size_t i) { return (n & (T(1) << i)) != 0; }
}
0