HaskellでFizzBuzz問題

(2007.6.1作成; 2007.6.3新規公開)

ちょー遅ればせながら、FizzBuzz問題について考えてみる。Rubyだと、考えるの一瞬、あとはタイピングの時間だけなので、つまらない。Haskellで考えてみよう。

1から100までの数をプリントするプログラムを書け。ただし3の倍数のときは数の代わりに「Fizz」と、5の倍数のときは「Buzz」とプリントし、3と5両方の倍数の場合には「FizzBuzz」とプリントすること。

こういう風に考えてみた;

  1. 1から100まで、ということは、まずリストを用意してみよう。
  2. リストの各要素について、(1) 数字を文字列にしたもの、あるいは (2) Fizz, Buzz, FizzBuzz に置きかえればいいんじゃないかな。
  3. あとは置き換え後のリストを連結すればいいだろう。

実装してみる。リストの各要素を変換するところはmapが使えそうだ。文字列のリストを連結して一つにまとめるのはunwordsかな。

(2017.3) コンパイル時の警告が出ないように修正。

Haskell
[RAW]
  1. repl_num :: Int -> String
  2. repl_num x
  3. | x `mod` 15 == 0 = "FizzBuzz"
  4. | x `mod` 3 == 0 = "Fizz"
  5. | x `mod` 5 == 0 = "Buzz"
  6. | otherwise = show x
  7. fizzbuzz :: [Int] -> String
  8. fizzbuzz list = unwords $ map repl_num list
  9. main :: IO ()
  10. main = do
  11. putStrLn $ fizzbuzz [1..100]

ghcでコンパイルし、実行してみる。

$ ./fizzbuzz-1
1 2 Fizz 4 Buzz Fizz 7 8 Fizz Buzz 11 Fizz 13 14 FizzBuzz 16 17 Fizz 19 Buzz Fizz 22 23 Fizz Buzz 26 Fizz 28 29 FizzBuzz 31 32 Fizz 34 Buzz Fizz 37 38 Fizz Buzz 41 Fizz 43 44 FizzBuzz 46 47 Fizz 49 Buzz Fizz 52 53 Fizz Buzz 56 Fizz 58 59 FizzBuzz 61 62 Fizz 64 Buzz Fizz 67 68 Fizz Buzz 71 Fizz 73 74 FizzBuzz 76 77 Fizz 79 Buzz Fizz 82 83 Fizz Buzz 86 Fizz 88 89 FizzBuzz 91 92 Fizz 94 Buzz Fizz 97 98 Fizz Buzz
いいようだ。

repl_num関数が冗長すぎる。タプルで設定を書いてみよう。その設定と置換対象の数値を受ける関数を書いてみる。

Haskell
[RAW]
  1. fizz = (3, "Fizz")
  2. buzz = (5, "Buzz")
  3. repl_num' t@(m, str) x
  4. | x `mod` m == 0 = str
  5. | otherwise = show x

でもこれだけじゃ15(の倍数)のときに困るなぁ。

Fizzに置換した場合でもBuzzを足すかどうか判定しないといけない、最後にFizzにもBuzzにもしなかった分だけ数値(を文字列にしたもの)を足すと。

fizz, buzzタプルをリストにして、変換対象のリストの各値について設定リストを順に見ていけばよさそうだ。こうかな?

Haskell
[RAW]
  1. -- ある値 x に対して, 変換の組のリストの先頭から, 順に割り切れるか試す.
  2. -- 複数のパタンがマッチすることがありうる.
  3. repl_num2 :: [(Int, String)] -> (Int -> String) -> Int -> String
  4. repl_num2 [] func x = func x
  5. repl_num2 ((m, str):xs) func x
  6. | x `mod` m == 0 = str ++ (repl_num2 xs nulstr x)
  7. | otherwise = repl_num2 xs func x
  8. where
  9. nulstr _ = "" -- どれかがマッチしたら、数字そのものは捨てる.
  10. fizzbuzz :: [Int] -> String
  11. fizzbuzz list = unwords $ map (repl_num2 [(3, "Fizz"), (5, "Buzz")] show) list
  12. main :: IO ()
  13. main = do
  14. putStrLn $ fizzbuzz [1..100]

動いているけど、どうもゴチャゴチャしてる。