Haskell - List(Data.Listモジュール, isPrefixOf関数)

すごいHaskellたのしく学ぼう!
(オーム社)
Miran Lipovača (著)
田中 英行 村主 崇行 (翻訳)
[Kindle版]

開発環境

• OS X Mavericks - Apple(OS)
• BBEdit - Bare Bones Software, Inc., Emacs (Text Editor)
• Haskell (純粋関数型プログラミング言語)
• GHC (The Glasgow Haskell Compiler) (処理系)
• The Haskell Platform (インストール方法、モジュール等)

初めてのコンピュータサイエンス(Jennifer Campbell、Paul Gries、Jason Montojo、Greg Wilson(著)長尾 高弘(翻訳))の12章(各種ツール)、12.7(練習問題)、12-20.をHaskellで解いてみる。

その他参考書籍

• プログラミングHaskell (オーム社) Graham Hutton(著) 山本 和彦(翻訳)
• Real World Haskell―実戦で学ぶ関数型言語プログラミング (オライリージャパン) Bryan O'Sullivan John Goerzen Don Stewart(著) 山下 伸夫 伊東 勝利 株式会社タイムインターメディア(翻訳)

12.7(練習問題)、12-20.

コード(BBEdit)

Sample.hs

{-# OPTIONS -Wall -Werror #-}
import Data.List

main :: IO ()
main = mapM_ print $ map (\(a, b, c, d) -> findButNotAfter a b c == d)
                         testCases

findButNotAfter :: String -> String -> String -> Maybe Int
findButNotAfter src pre pattern = findButNotAfter' src pre pattern 0

findButNotAfter' :: String -> String -> String -> Int -> Maybe Int
findButNotAfter' _ [] [] _ = Nothing
findButNotAfter' _ [] _ _ = Nothing
findButNotAfter' _ _ [] _ = Nothing
findButNotAfter' [] _ _ _ = Nothing
findButNotAfter' src pre pattern n =
    let i = findStr src pattern 0
    in if i == -1 then
           Nothing
       else
           let s = take i src
           in if s == pre then
                  let l = n + i + i
                  in findButNotAfter' (drop l src) pre pattern l
              else
                  Just (n + i)

findStr :: String -> String -> Int -> Int
findStr [] _ _ = -1
findStr src@(_:xs) pattern n = if isPrefixOf pattern src then
                                   n
                               else
                                   findStr xs pattern (n + 1)

testCases :: [(String, String, String, Maybe Int)]
testCases = [("", "", "", Nothing), ("", "", "a", Nothing),
             ("", "a", "", Nothing), ("", "a", "b", Nothing),
             ("a", "", "b", Nothing), ("a", "a", "", Nothing),
             ("abcd", "a", "bc", Nothing), ("abcdcd", "ab", "cd", Just 4),
             ("abxcdcd", "ab", "cd", Just 3)]

入出力結果(Terminal, runghc)

$ runghc Sample.hs
True
True
True
True
True
True
True
True
True
$

慣れるまでは{-# OPTIONS -Wall -Werror #-}の記述を消さずに細かく型を指定していくことに。