あみだくじ in Haskell DouKaku?

次のような書式で与えられた「あみだくじ」があります。
(あみだくじはコード中に埋め込んでも、標準入力や
外部ファイルから読み込んでも、書きやすい方法でかまいません)

A B C D E
| | |-| |
|-| | |-|
| |-| |-|
|-| |-| |
|-| | | |

このあみだくじをたどって
A B C D E
| | |-| |
|-| | |-|
| |-| |-|
|-| |-| |
|-| | | |
B D C A E
のように結果を表示させるプログラムを作ってください。

どうでしょう

s = ["A B C D E",
     "| | |-| |",
     "|-| | |-|",
     "| |-| |-|",
     "|-| |-| |",
     "|-| | | |"]

repl (x:' ':xs) ('|':y:ys) =
    let z:zs = repl xs ys in
    case y of
      '-' -> z:' ':x:zs
      _ -> x:' ':z:zs
repl xs _ = xs

main = putStrLn $ (unlines s) ++ foldl1 repl s

初心者なりにがんばってみました。出来上がってからMinkeさんの回答を見てアゴが勢いよく机に落ちました…　まさしくエレガント。

大変勉強になりました。

module Main
    where

import System (getArgs)
import Data.List (intersperse)

swaps :: String -> String -> String
swaps (px1:px2:pxs) (pl1:pl2:pls)
    | pl1 == '-' = [px2, px1] ++ swaps pxs pls
    | otherwise = [px1] ++ swaps (px2:pxs) (pl2:pls)
swaps xs _ = xs

followLines:: String -> String -> String
followLines pls line = swaps pls (filter (/= '|') line)

solveAmida :: [String] -> String
solveAmida (x:xs) = intersperse ' ' $ foldl (followLines) (filter (/= ' ') x) xs
    
main :: IO()
main = do
    args <- getArgs
    body <- readFile $ head args
    putStrLn body
    putStrLn $ solveAmida $ lines body

DouKaku?

あみだくじ

Posted feedbacks - Haskell

Index

Feed

Other

Link

Pathtraq