如何在Haskell中为二叉树实现类Unix tree风格的Show实例

嘿，我来帮你搞定这个按层级缩进的二叉树Show实例问题！你想要让自定义的二叉树：

data Tree a = Nil | Leaf a | Branch a (Tree a) (Tree a)

输出像Unix tree命令那样的层级结构，比如：

27

14

10
19
35
31
42

但目前的代码只能固定加一个制表符缩进，没法根据子树深度动态调整。别担心，我们只需要给递归过程加个深度参数就能解决这个问题！

核心思路：用辅助函数传递深度 #

问题出在你当前的show函数没有记录层级深度——每次递归都只加一个\t，不管子树有多深。我们可以写一个带深度参数的辅助函数，让它根据当前层级生成对应的缩进，然后让主show方法调用这个辅助函数的初始状态（根节点深度为0）。

完整实现代码 #

这里我用>>作为每层的缩进符号（和你的示例格式匹配），你也可以换成制表符\t，调整起来很简单：

data Tree a = Nil | Leaf a | Branch a (Tree a) (Tree a)

import Data.List (intercalate)

instance (Show a) => Show (Tree a) where
    show tree = showTree tree 0
      where
        -- 辅助函数：接收树节点和当前层级深度
        showTree Nil _ = ""
        showTree (Leaf e) depth = replicate depth '>' ++ " " ++ show e
        showTree (Branch e ls rs) depth = 
            let currentLine = replicate depth '>' ++ " " ++ show e
                -- 子节点深度比当前节点深一级
                leftSubtree = showTree ls (depth + 1)
                rightSubtree = showTree rs (depth + 1)
                -- 过滤掉空字符串（比如Nil的情况），避免多余换行
                validChildren = filter (not . null) [leftSubtree, rightSubtree]
            in -- 如果有子节点，就换行拼接；没有的话直接返回当前行
               currentLine ++ if null validChildren then "" else "\n" ++ intercalate "\n" validChildren

代码细节解释 #

1. 深度参数的作用：根节点深度设为0，每往下递归一层深度+1，用replicate depth '>'生成对应数量的缩进符号，比如深度2就生成>>。
2. 处理空节点：Nil返回空字符串，并且在拼接子节点时过滤掉空串，这样不会出现多余的空行。
3. 换行处理：用intercalate "\n"把非空的子节点字符串用换行连接，保证每个子节点单独占一行。

换成制表符缩进的版本 #

如果你更习惯用制表符\t，只需要把缩进生成的部分改成replicate depth '\t'就行，比如：

showTree (Leaf e) depth = replicate depth '\t' ++ show e
showTree (Branch e ls rs) depth = 
    let currentLine = replicate depth '\t' ++ show e
        -- ... 其他部分不变

测试一下 #

构建你示例里的树：

testTree :: Tree Int
testTree = Branch 27
            (Branch 14 (Leaf 10) (Leaf 19))
            (Branch 35 (Leaf 31) (Leaf 42))

运行print testTree，就能得到和你期望完全一致的输出啦！

内容的提问来源于stack exchange，提问作者Juan Ortega