結果

問題 No.58 イカサマなサイコロ
コンテスト
ユーザー wotsushi
提出日時 2017-04-15 11:09:55
言語 Ruby
(3.4.1)
結果
AC  
実行時間 99 ms / 5,000 ms
コード長 771 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 92 ms
コンパイル使用メモリ 7,680 KB
実行使用メモリ 12,288 KB
最終ジャッジ日時 2024-07-18 17:53:35
合計ジャッジ時間 1,643 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #
raw source code 

N = gets.to_i
K = gets.to_i
def f(n, m)
  @dp ||= {}
  @dp[[n, m]] ||= if n == 0 and m == 0
                    1.0
                  elsif n == 0 or m <= 0
                    0.0
                  elsif n <= K
                    (4..6).map {|i| f(n - 1, m - i)}.inject(:+) / 3
                  else
                    (1..6).map {|i| f(n - 1, m - i)}.inject(:+) / 6
                  end
end
def g(n, m)
  @dp2 ||= {}
  @dp2[[n, m]] ||= if n == 0 and m == 0
                    1.0
                  elsif n == 0 or m <= 0
                    0.0
                  else
                    (1..6).map {|i| g(n - 1, m - i)}.inject(:+) / 6
                  end
end
ans = ((N + 1)..(6 * N)).map {|m| f(N, m) * (N...m).map {|l| g(N, l)}.inject(:+)}.inject(:+)
puts ans
0