Haskell代数数据类型模式匹配

时间:2012-05-04 22:47:47

标签: haskell pattern-matching algebraic-data-types

我有以下内容:

data Alpha a = Beta a [Alpha a]
val = Beta 1 [Beta 2 [], Beta 5 [Beta 7 []]]

我正在尝试定义一个将在Alpha Int类型的v​​al上移动并对其求和的函数。我想要的方法是提取所有的Ints,然后对结果列表求和,但我正在努力提取所有的Int,因为我不知道如何处理递归...

轻微的尝试:

checkAlpha :: Alpha Int -> [Int]
checkAlpha (Beta a []) = [a]
checkAlpha (Beta a b) = [a] ++ (map checkAlpha b)

显然这不起作用,但我看不到解决方案。

2 个答案:

答案 0 :(得分:7)

如果您使用

concatMap :: (a -> [b]) -> [a] -> [b]

而不是map,它会起作用并且足够优雅。

您不需要将空列表的情况特别视为第二个组件,

checkAlpha :: Alpha a -> [a]
checkAlpha (Beta a alphas) = a : concatMap checkAlpha alphas

执行您想要的操作,并且与参数类型无关。

答案 1 :(得分:0)

您可以考虑使用Tree代替Alpha,它有许多方便的操作:

> flatten $ Node 1 [Node 2 [], Node 5 [Node 7 []]]
[1,2,5,7]