第9回オフラインリアルタイムどう書くの参考問題をHaskellで書いた

第9回オフラインリアルタイムどう書くの参考問題をHaskellで書きました。

オフラインリアルタイムどう書くの過去問でHaskellスキルアップを目指します。

makeNumber.hs

import Data.List
import Control.Applicative
import Text.Regex

--先頭が0でない長さ4の順列
permutations' :: [Int] -> [[Int]]
permutations' ns = nub $ filter (\x -> head x /= 0) $ map (take 4) $ permutations ns

--6を9に変換する
convert6To9 :: [Int] -> [[Int]]
convert6To9 [] = [[]]
convert6To9 (n:ns) | n /= 6 = map (n:) $ convert6To9 ns
				   | otherwise = [(n:), (9:)] <*> convert6To9 ns

make4DigitsNumber :: [Int] -> [[Int]]
make4DigitsNumber ns = sort $ concatMap convert6To9 $ permutations' ns

listToString :: [Int] -> String
listToString [] = ""
listToString (n:ns) = show n ++ listToString ns

solve :: Int -> [Int] -> String
solve n ns = let list = make4DigitsNumber ns in
				if length list >= n
				then listToString $ list !! (n - 1)
				else "-"

parseTestData :: String -> [(Int, [Int], String)]
parseTestData str = map parseLine $ lines str
	where
		parseLine :: String -> (Int, [Int], String)
		parseLine line = let [input, output] = words line
		                     (index, numList) = parseInput input
		                 in (index, numList, output)
		parseInput :: String -> (Int, [Int])
		parseInput input = let [index, numList] = splitRegex (mkRegex ":") input
						   in (read index :: Int, parseNumList numList)
		parseNumList :: String -> [Int]
		parseNumList [] = []
		parseNumList (n:ns) = (read (n:[]) :: Int):(parseNumList ns)

main = do
	testData <- parseTestData <$> getContents
	mapM (\(index, numList, output) -> print $ (solve index numList) == output) testData