使用Haskell打印矩阵

时间:2012-10-27 17:06:01

标签: haskell matrix

我需要在haskell中打印出一个矩阵,所以看起来像这样:

main> putStr (showMat [[1,-500,-4], [100,15043,6], [5,3,10]])
       1  -500 -4
     100 15043  6
       5     3 10

到目前为止,我已经想出了这个:

type Matrix a = [[a]]
type IntMat = Matrix Integer

showMat :: IntMat -> String
showMat [] = ""
showMat ((y:ys):xs)  = (printRow (rowmaxs) (elements) (y:ys)) ++ "\n" ++ showMat xs
  where rowmaxs = rowMaxs ((y:ys):xs) ; elements = elementLengths (y:ys)

rowMaxs :: IntMat -> [Int]
rowMaxs [] = []
rowMaxs (x:xs) = [length (show (maximum (x)))] ++ (rowMaxs xs)

elementLengths :: [Integer] -> [Int]
elementLengths [] = []
elementLengths (y:ys) = [length (show y)] ++ (elementLengths ys)

printRow :: [Int] -> [Int] -> [Integer] -> String
printRow [] (a:as) (y:ys) = ""
printRow (z:zs) (a:as) [] = ""
printRow [] [] (y:ys)     = ""
printRow [] [] []         = ""
printRow (z:zs) (a:as) (y:ys) = addSpaces (z-a) ++ show y ++ [' '] ++ printRow zs as ys

addSpaces :: Int -> String
addSpaces 0 = ""
addSpaces n = " " ++ addSpaces (n-1)

返回此内容:

Main> putStr (showMat [[1,23,1],[23,56,789],[1234,0,1]])
      1  23    1 
      23   56 
     1234

我可以看到行末的元素丢失是由于printRow函数中的情况,但我不知道如何修复它。字符也没有考虑到它们之前印刷的字符。任何有关我如何解决这个问题的帮助将不胜感激。

2 个答案:

答案 0 :(得分:2)

前段时间我写了一个小脚本,它采用了一个以制表符分隔的表格(CSV样式)并将其打印到控制台。我已经提取了相关部分并上传了源代码here。例如,给定您的输入

m = [[1,23,456],[78,-90,123],[4567,8,9]]

putStrLn $ showMat m将进入

  1  -500 -4
100 15043  6
  5     3 10

如果你想要左对齐,那么最底层还有一个功能(注释掉)。

答案 1 :(得分:1)

showMat中,您为每行重新计算rowmaxselements。特别是,rowmaxs在矩阵中有每个行左的元素,但printRow使用它作为矩阵的每个的含义

修改:这是showMat稍微好一点的版本:

showMat :: IntMat -> String
showMat rows = concat (map perRow rows)
  where rowmaxs = rowMaxs rows
        perRow cols = printRow rowmaxs elements cols ++ "\n"
          where elements = elementLengths cols

它仍然是错误的--- rowMaxs(试图)计算每行中数字的最大长度,但你真的想要每列中数字的最大长度。这样做的一个结果是你偶尔会将负数传递给addSpaces,而这个数字并不能很好地应对,所以这里的addSpaces版本表现得更优雅:

addSpaces :: Int -> String
addSpaces n = replicate n ' '

现在我们得到了这个:

*Main> putStr (showMat [[1,23,1],[23,56,789],[1234,0,1]])
 1  23    1 
23  56  789 
1234   0    1 

更好,但尚未正常工作。

我没有修复代码中的所有错误,因为我觉得你还在学习,需要亲自找到它们的经验。

我建议使用map / zipWith / zipWith3而不是显式编写递归,因为它使代码更容易理解。