Haskell Tree 删除一个节点/计算所有可能的路径

Question

我有这些结构：

data Tree = Leaf Points | Branch Points [Tree]
deriving(Eq,Show)

data Design = Design Tree Tree Int
deriving(Eq,Show)

数据类型Tree是数据类型Design的一部分，Design拥有两个Trees：

 type Points = [(Int,Int,Int,Int)]

 design = Design mtree btree 180

我必须找到所有可能的路径：

numberPaths :: Design -> Int

并将第三个 Integer 从 Points 更改为 180 从 design：

type Points = [(Int,Int,Int,Int)]
design = Design mtree btree 180

changeInt :: Design -> Design

Answer 1

计算路径

为了找到一棵树中的路径数，我们递归地沿着树向下计算每个节点的子路径数，然后将它们相加。所以除了numberPaths作为接口之外，我们还需要另一个辅助函数，其类型签名是Tree -> Int。我们就叫它descend吧。

descend :: Tree -> Int 

我们将 Leaf 视为 1，将 Branch 视为其子树的路径数之和。所以descend 变成：

descend :: Tree -> Int 
descend (Leaf _)            = 1 
descend (Branch _ subtress) = foldr (\tree sum -> sum + descend tree) 0 subtrees 

函数numberPaths只需要调用descend就可以计算出它的两棵树。

numberPaths :: Design -> Int 
numberPaths (Design a b _) = descend a + descend b 

改变诠释

changeInt的类型签名表示没有引入Int值来改变Design类型。我认为这没有意义，所以我冒昧地将其更改为Int -> Design -> Design。

changeInt 的实现遵循类似的模式。为了改变树，我们递归地遍历树的每一层。我们还引入了一个辅助函数changePoints。

changeTree :: Int -> Tree -> Tree
changeTree n (Leaf points)            = Leaf $ changePoints n points
changeTree n (Branch points subtrees) = Branch points' subtrees'
    where points'   = changePoints n points 
          subtrees' = map (changeTree n) subtrees

changePoints :: Int -> Points -> Points 
changePoints n points = map (\(x, y, _, z) -> (x, y, n, z)) points 

changeInt 然后只需调用changeTree 即可完成工作。

changeInt :: Int -> Design -> Design 
changeInt n (Design mtree btree _) = Design mtree' btree' n
    where mtree' = changeTree n mtree 
          btree' = changeTree n btree