树在 Haskell 中工作

Question

我正在做一个项目并且遇到了一些困难，因为我不习惯 Haskell，而且它对我来说是一种很难编码的语言。

我想要一棵有叶子和树枝的树，叶子只能作为整数输入，比如 1,2,3。 分支将仅为 operation1 和 operation2 的字符。

这些操作从叶子中获取输入并进行计算并将输出发送到其他分支并再次进行计算，直到它到达顶部。

到目前为止我的代码：

data Tree = Leaf Float | Branch Char Tree Tree deriving (Show,Eq,Ord)

insertElement x (Leaf y) = Leaf y

insertElement x (Branch a left right) = Branch a (insertElement x left) right                                                           
doIt (x:xs) (y:ys) =  insertElement ((Branch y (Leaf x) (Leaf (head xs))))                                  

op1 num1 num2 = 1+num1+num2*2
op2 num1 num2 = 1-num1*num2*2

树将被递归创建，我尝试使用 foldr 来执行此操作但没有工作，因为它不是二叉树。

任何帮助都会很好。

Answer 1

我不确定我是否理解这个问题，但也许你想要这样的东西：

data Op = Op1 | Op2
data Tree = Leaf Float | Branch Op Tree Tree

eval :: Tree -> Float
eval (Leaf x) = x
eval (Branch Op1 l r) = op1 (eval l) (eval r)
eval (Branch Op2 l r) = op2 (eval l) (eval r)

op1 :: Float -> Float -> Float
op1 num1 num2 = 1+num1+num2*2
op2 :: Float -> Float -> Float
op2 num1 num2 = 1-num1*num2*2

example :: Float
example = eval (Branch Op1 (Leaf 3) (Branch Op2 (Leaf 2) (Leaf 1)))

如果您想这样做，请考虑使用比Tree 更通用的类型，例如

data Exp = Lit Float | Op1 Exp Exp | Op2 Exp Exp
eval :: Exp -> Float
eval (Lit x) = x
eval (Op1 l r) = op1 (eval l) (eval r)
...