結果

問題 No.58 イカサマなサイコロ
コンテスト
ユーザー ducktail
提出日時 2018-07-17 15:03:53
言語 Haskell
(9.10.1)
結果
AC  
実行時間 3 ms / 5,000 ms
コード長 860 bytes
コンパイル時間 11,623 ms
コンパイル使用メモリ 205,340 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-25 00:58:30
合計ジャッジ時間 12,494 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.8.2/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff #
raw source code 

import Control.Applicative ((<$>), (<*>))
import Control.Monad (guard)
import Data.List (foldl')
import Data.Vector.Unboxed (Vector, (!), (//))
import qualified Data.Vector.Unboxed as V

main :: IO ()
main = solve <$> readLn <*> readLn >>= print

solve :: Int -> Int -> Double
solve n k = sum $ do
  ti <- [0 .. 6*n]
  ji <- [0 .. 6*n]
  guard $ ti > ji
  return $ dpt ! ti * dpj ! ji
  where dpt = foldl' f ((V.replicate (6*n+1) 0) // [(0,1.0)]) ((replicate (n-k) l1) ++ (replicate k l2))
        dpj = foldl' f ((V.replicate (6*n+1) 0) // [(0,1.0)]) (replicate n l1)
        f dp ds = V.accum (+) (V.replicate (6*n+1) 0) $ do
          (v, p) <- ds
          i <- [0..6*n]
          guard $ i >= v
          return (i, dp ! (i-v) / p)

l1 = [(1,6.0),(2,6.0),(3,6.0),(4,6.0),(5,6.0),(6,6.0)] :: [(Int,Double)]
l2 = [(4,3.0),(5,3.0),(6,3.0)] :: [(Int,Double)]
0