从haskell中的输入文件读取

Question

我正在解决Project Euler Problem 11。我已将问题中的表格复制粘贴到名为“input.txt”的文件中。输入文件的每一行包含一行 20x20 矩阵，列之间用空格分隔。

我想要的是一个读取这个文件并将它作为一个 IO 数组返回的函数。我在这样做时遇到了很大的麻烦。

到目前为止，我已经这样做了：

import System.IO
import Control.Monad

main = readFile "input.txt"

这当然只给了我输入文件的 IO 字符串表示，但我尝试的一切似乎都失败了。我应该如何进行？我知道我应该做类似的事情

array ((1,1),(20,20)) [ the numbers tupled with their indices ]

但是对我来说转换数字是完全不可能的，很可能是因为我还没有完全理解单子。

我相信这实际上很容易，一旦你理解了它。

有没有人有什么建议？

Answer 1

你可以用这样的方式解析你的文件：

s = "2 4\n6 8"

s' :: [Int]
s' = (map read . words) s

let arr = listArray ((1,1),(2,2)) s'
-- arr == array ((1,1),(2,2)) [((1,1),2),((1,2),4),((2,1),6),((2,2),8)]

通过将words 与map read（其中read :: (Read a) => String -> a）结合起来，您将得到一个[Int] 列表。

因此，为了稍微组织一下，您在 IO monad 中的代码可能如下所示（假设每一行都有相同的固定列数，并且您获取输入文件名、行数和列作为命令行参数）：

module Main

where

import Data.Array
import Control.Monad
import System.Environment

readWords :: (Read a) => String -> [a]
readWords = map read . words

parseFile :: String -> Int -> Int -> IO (Array (Int, Int) Int)
parseFile fname rows cols = do
    matr <- liftM readWords $ readFile fname
    return $ listArray ((1, 1), (rows, cols)) matr
    -- (matr :: [Int] is inferred from the parseFile's type)

main :: IO ()
main = do
    args <- getArgs
    case args of
        [fname, rows, cols] -> do
            arr <- parseFile fname (read rows) (read cols)
            print arr

请注意转换函数readWords 是如何能够转换为任何[a] 的，前提是Read a，这样我们就不会仅限于整数。 liftM 函数采用纯函数（我们的 readWords）并“提升”它以在当前 monad 中工作，即 IO。

Answer 2

整个程序应该是这样的：

import System.IO
import Data.Array

main = do
    content <- readFile "input.txt"
    let matrix = map (map read . words) $ lines content     -- [[Int]]
    let arr = listArray ((1, 1), (20, 20)) (concat matrix)
    putStr $ solve arr

其中solve是Array (Int, Int) Int -> String类型的函数。