DouKaku? 6835 mattsan: きっともっと綺麗な書き方があると思うので...(2次元ランダムウォーク)

Comment detail

きっともっと綺麗な書き方があると思うのですが、挑戦の意味を込めて投稿です。

import System.Random
import Text.Printf

direction = [(1, 0), (0, 1), (-1, 0), (0, -1)]

printPoint :: (Int, (Int, Int)) -> IO ()
printPoint (n, (x, y)) = printf "%4d%4d%4d¥n" n x y

addPair (x1, y1) (x2, y2) = (x1 + x2, y1 + y2)

main = do
  gen <- getStdGen
  mapM_ printPoint $ take 10 $ zip [1..] $ scanl addPair (0, 0) $ map (\x -> direction !! (mod x 4)) $ randoms gen

僕もhaskellの練習で作ってみました。長くなってしまった。乱数の使い方が難しくって強引な事(unsafeを利用している）をしたけど、mattanさんがやってるようなすっきりとした利用方法があるんですね。

>writeRW2D "file.name" 100

時刻0-99までのxy位置が書かれる。

import System.IO.Unsafe 
import System.Random

type XYPlot a = (a,a)

random4:: IO Int
random4 = getStdRandom (randomR (0,3))

nextStep::  XYPlot Int -> IO Int -> XYPlot Int 
nextStep (x,y) iorand 
    | rand == 0 = (x+1,y)
    | rand == 1 = (x,y+1)
    | rand == 2 = (x-1,y)
    | rand == 3 = (x,y-1)
    where rand =  unsafePerformIO $ iorand

randomWalk2D:: XYPlot Int -> [XYPlot Int]
randomWalk2D (x,y) = [(x,y)]++randomWalk2D (nextStep (x,y) random4)

formatRW2D:: XYPlot Int -> String
formatRW2D (x,y) = (show x) ++ " " ++ (show y) ++ "\n"

rw2DtoStr:: Int -> [XYPlot Int]->String
rw2DtoStr count (x:xs) = show count ++ " " ++
                         formatRW2D x ++ (rw2DtoStr (1+ count) xs)
rw2DtoStr count [] = ""

writeRW2D::FilePath -> Int ->IO()
writeRW2D fname terminal =      
    writeFile fname $rw2DtoStr 0 $take terminal $randomWalk2D (0,0)

DouKaku?

Comment detail

Index

Feed

Other

Link

Pathtraq