練習問題

「入門Haskell―はじめて学ぶ関数型言語」　p.101 より。

ところで，あるリストに複数の型を含めたい場合にはどうしたらいいか，という話ですが，そういう場合にも data は便利です。たとえば次のように
data MyType = I Int | F Float | C Char
ここでたとえば [I 10, F 10, C 'a'] のようなリストがあるとします。

(1)[MyType] から C 要素だけ取り出す関数 filterC を書きなさい。

これは C 要素であるか否かを判定する関数を作って filter すればいいんだろう。判定には case を使って場合分けすればいい。

 filterC :: [MyType] -> [MyType]
 filterC = filter isC
   where isC x = case x of
                   I a -> False
                   F a -> False
                   C a -> True

結果。

 *Main> filterC [I 10, F 10, C 'a']
 [C 'a']

OK。

(2) [MyType] から I 要素だけを取り出してその合計を計算する sumI を書きなさい。

 sumI :: [MyType] -> Int
 sumI = sum . map pickI
   where pickI x = case x of
                     I a       -> a
                     otherwise -> 0

 *Main> sumI [I 10, F 10, C 'a']
 10

なるほど，otherwise も使えるんだな。

(3) [MyType] から，I 要素はそのまま，F 要素は整数にし（truncate 関数が使えます），C 要素はその文字の文字コードを計算し（ Data.Char をインポートすれば ord 関数が計算してくれます），その合計を計算する mySum 関数を書きなさい。

 import Data.Char
 
 mySum :: [MyType] -> Int
 mySum = sum . map toInt
   where toInt x = case x of
                     I a -> a
                     F a -> truncate a
                     C a -> ord a

 *Main> mySum [I 10, F 10, C 'a']
 117

'a' のコードは 97 だから全部足すと 117。OK。

ところで，GHCi で表示できるように，MyType の宣言は次のようにしている。

 data MyType = I Int | F Float | C Char deriving (Show)