計算機科学のブログ

関数型プログラミングの基礎 ファーストクラス関数 引数としての関数 例:カスタムソート Data.Listモジュール、sortBy関数

入門Haskellプログラミング (Will Kurt(著)、株式会社クイープ(監修、翻訳)、翔泳社)のUNIT1(関数型プログラミングの基礎)、LESSON4(ファーストクラス関数)、4.1(引数としての関数)、例:カスタムソートのクイックチェック 4-2の解答を求めてみる。

コード

import Data.List

compareNames name1 name2 =
  let lastName1 = snd name1
      lastName2 = snd name2
   in if lastName1 > lastName2
        then GT
        else
          if lastName1 < lastName2
            then LT
            else
              let firstName1 = fst name1
                  firstName2 = fst name2
               in if firstName1 > firstName2
                    then GT
                    else
                      if firstName1 < firstName2
                        then LT
                        else EQ

names =
  [ ("Ian", "Curtis"),
    ("Bernard", "Sumner"),
    ("Peter", "Hook"),
    ("Stephen", "Morris"),
    ("Bernard", "Hook")
  ]

main :: IO ()
main = do
  mapM_ print $ sortBy compareNames names

入出力結果(Terminal, Zsh)

% runghc sample02.hs 
("Ian","Curtis")
("Bernard","Hook")
("Peter","Hook")
("Stephen","Morris")
("Bernard","Sumner")
%