計算機科学のブログ

実践Haskell 効率的でステートフルな配列 バブルソートを実装する UArray、Data.Array.Unboxedモジュール、UArray型

入門Haskellプログラミング (Will Kurt(著)、株式会社クイープ(監修、翻訳)、翔泳社)のUNIT7(実践Haskell)、LESSON 42(Haskellでの効率的でステートフルな配列)、42.4(バブルソートを実装する)、UArrayを更新するのクイックチェック 42-4の解答を求めてみる。

コード

import Control.Monad
import Control.Monad.ST
import Data.Array.ST
import Data.Array.Unboxed

listToUArray :: [Int] -> UArray Int Int
listToUArray vals = runSTUArray $ do
  let end = length vals - 1
  myArray <- newArray (0, end) 0
  forM_ [0 .. end] $ \i -> do
    let val = vals !! i
    writeArray myArray i val
  return myArray

myData :: UArray Int Int
myData = listToUArray [7, 6, 4, 8, 10, 2]

main :: IO ()
main = do
  print myData

入出力結果(Terminal, Zsh)

% runghc sample04.hs 
array (0,5) [(0,7),(1,6),(2,4),(3,8),(4,10),(5,2)]
%