計算機科学のブログ

関数型プログラミングの基礎 関数と関数型プログラミング if/then/else、even関数、剰余関数、mod関数

入門Haskellプログラミング (Will Kurt(著)、株式会社クイープ(監修、翻訳)、翔泳社)のUNIT1(関数型プログラミングの基礎)、LESSON 2(関数と関数型プログラミング)、2.5(練習問題)Q2-3の解答を求めてみる。

コード

lesson/app/Main.hs

module Main where

import Lib

f1 :: Integer -> Integer
f1 n =
  if even n
    then n - 2
    else 3 * n + 1

f2 :: Integer -> Integer
f2 n =
  if n `mod` 2 == 0
    then n - 2
    else 3 * n + 1

fs :: [Integer -> Integer]
fs = [f1, f2]

ns :: [Integer]
ns = [-4 .. 5]

main :: IO ()
main = do
  mapM_
    ( \f ->
        mapM_
          (print . f)
          ns
    )
    fs

入出力結果(Terminal, Zsh)

% stack runghc app/Main.hs
-6
-8
-4
-2
-2
4
0
10
2
16
-6
-8
-4
-2
-2
4
0
10
2
16
%