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



这里有一个Haskell函数,它接受一个数字n并返回第n个fibonacci数。(我使用了索引方案,第0个数字是0,第1个数字是1,第2个数字是一,第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的惰性求值,强制对更新后的参数进行求值(例如,使用$!运算符?)最优雅/最惯用的方法是什么?

您可以使用bang模式来实现这一点。

{-# 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

最新更新