Haskell - 如何在尾递归函数中更新参数时绕过延迟求值

时间:2012-08-19 15:58:31

标签: haskell pattern-matching lazy-evaluation fibonacci tail-recursion

这是一个Haskell函数,它接受一个数字n并返回第n个斐波纳契数。 (我使用了索引方案,使得第0个数字为0,第1个数字为1,第2个数字为1,第3个数字为2,依此类推。)

fib :: (Integral a) => a -> a
fib 0 = 0
fib n = fibhelper n 0 1

fibhelper :: (Integral a) => a -> a -> a -> a
fibhelper 1 x y = y
fibhelper n x y = fibhelper (n-1) y (x+y)

现在,假设为了效率,我想绕过Haskell的懒惰评估并强制评估更新的参数(例如使用$!运算符?)什么是最优雅的/惯用这种方式吗?

1 个答案:

答案 0 :(得分:8)

您可以使用爆炸模式来执行此操作。

{-# LANGUAGE BangPatterns #-}

fib :: (Integral a) => a -> a
fib 0 = 0
fib n = fibhelper n 0 1

-- Everything with a ! before it will be evaluated before the function body
fibhelper :: (Integral a) => a -> a -> a -> a
fibhelper 1 _ (!y) = y
fibhelper (!n) (!x) (!y) = fibhelper (n-1) y (x+y)
-- The above line is equivalent to:
--fibhelper n x y = n `seq` x `seq` y `seq` fibhelper (n-1) y (x+y)

另请注意,使用Integral类型类时,您有点过于热心。你真的希望斐波那契数列的索引与值的类型相同吗?我建议您改为使用签名:

fib :: (Integral a, Integral b) => a -> b

此外,如果您正在寻找性能,则应完全避免使用Integral