結果

問題 No.58 イカサマなサイコロ
ユーザー hytkxdhytkxd
提出日時 2023-09-02 20:30:30
言語 Ruby
(3.4.1)
結果
AC  
実行時間 101 ms / 5,000 ms
コード長 1,021 bytes
コンパイル時間 259 ms
コンパイル使用メモリ 7,296 KB
実行使用メモリ 12,416 KB
最終ジャッジ日時 2024-06-12 02:51:40
合計ジャッジ時間 1,780 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

#!/usr/local/bin/ruby
class FakeDice
  def initialize(*arg)
    @n,@k = arg
    @jirodice = normal_dice_pb(@n)
    @tarodice = fake_dice_pb(@k,normal_dice_pb(@n-@k))
  end

  private
  def normal_dice_pb(n)
    dp = Hash.new(0)
    dp[0]=1
    n.times do
      newdp = Hash.new(0)
      dp.each do |dsum,pb|
        (1..6).each do |i|
          newdp[dsum+i] += pb/6.to_r
        end
      end
      dp = newdp
    end
    dp
  end
  def fake_dice_pb(k,dp)
    k.times do
      newdp = Hash.new(0)
      dp.each do |dsum,pb|
        (4..6).each do |i|
          newdp[dsum+i] += pb/3.to_r
        end
      end
      dp = newdp
    end
    dp
  end

  public

  def ans
    cumsum = Array.new(@n*6+2,0)
    d = @n*6
    while d >= @n+1
      cumsum[d] = cumsum[d+1] + @tarodice[d]
      d-=1
    end
    r = 0
    d =@n
    while d < @n*6
      r += @jirodice[d]*cumsum[d+1]
      d+=1
    end
    r.to_f
  end
end
### END: class FakeDice
iod = STDIN
n = iod.gets.to_i
k = iod.gets.to_i
puts FakeDice.new(n,k).ans
exit 0
0