enumFromTo如何在haskell中工作以及哪些优化加速了GHC实现与天真实现的加速

时间:2018-03-06 05:44:13

标签: haskell

我正在学习haskell,其中一项练习要求我写一个等同于enumFromTo的函数。

我想出了以下两个实现:

eft' :: Enum a => a -> a -> [a]
eft' x y = go x y []
  where go a b sequence
          | fromEnum b < fromEnum a = sequence
          | otherwise = go (succ a) b (sequence ++ [a])

eft :: Enum a => a -> a -> [a]
eft x y = go x y []
  where go a b sequence
          | fromEnum b < fromEnum a = sequence
          | otherwise = go a (pred b) (b : sequence)

我有一种预感,第一个版本做得更多,因为它将每个元素放入列表并连接到现有的sequence,而第二个版本将单个元素添加到列表中。这是性能差异的主要原因还是有其他重要因素,还是我的预感略有偏差?

使用:set +s在ghci中进行测试显示在我的机器上(Windows 10,GHC 8.2.2,intel i7-4770HQ):

*Lists> take 10 (eft 1 10000000)
[1,2,3,4,5,6,7,8,9,10]
(9.77 secs, 3,761,292,096 bytes)
*Lists> take 10 (eft' 1 10000000)
[1,2,3,4,5,6,7,8,9,10]
(27.97 secs, 12,928,385,280 bytes)
*Lists> take 10 (enumFromTo 1 10000000)
[1,2,3,4,5,6,7,8,9,10]
(0.00 secs, 1,287,664 bytes)

我的第二个预感是take 10 (eft 1 10000000)应该比take 10 (eft' 10000000)表现更好,因为后者必须从10000000到10之间一直建立列表才能返回任何有用的值take 1}}。显然这种预感是错误的,我希望有人可以解释原因。

最后,ghc实现比我的天真实现更高效。我很想知道已经应用了其他优化来加速它。这个similarly titled SO question的答案分享了一些似乎来自ghc实现的代码,但没有解释“肮脏”如何提高效率。

1 个答案:

答案 0 :(得分:5)

eft的问题在于它仍然需要构建整个列表,无论您是否尝试使用take 10将其删除。 Tail recursion is not your friend when you want to build things lazily。你想要的是guarded recursion(即在foldr之后的相关构造函数后面的递归调用,以便在你不需要它们时可以不加评价):

eft'' :: Enum a => a -> a -> [a]
eft'' x y
    | fromEnum y < fromEnum x = []
    | otherwise = x : eft'' (succ x) y
GHCi> take 10 (eft 1 10000000)
[1,2,3,4,5,6,7,8,9,10]
(7.48 secs, 2,160,291,096 bytes)
GHCi> take 10 (eft'' 1 10000000)
[1,2,3,4,5,6,7,8,9,10]
(0.00 secs, 295,752 bytes)
GHCi> take 10 (enumFromTo 1 10000000)
[1,2,3,4,5,6,7,8,9,10]
(0.00 secs, 293,680 bytes)

至于eft'eft更差,这确实与(++)有关。作为参考,以下是take(++)的定义(我使用the Report definitions而不是the GHC ones,但这里的细微差别实际上并不重要):

take                   :: Int -> [a] -> [a]  
take n _      | n <= 0 =  []  
take _ []              =  []  
take n (x:xs)          =  x : take (n-1) xs 

(++) :: [a] -> [a] -> [a]  
[]     ++ ys = ys  
(x:xs) ++ ys = x : (xs ++ ys)

如果你手工评估eft,你会看到在给你任何元素之前如何构建整个列表:

take 3 (eft 1 5)
take 3 (go 1 5 [])
take 3 (go 1 4 (5 : []))
take 3 (go 1 3 (4 : 5 : []))
-- etc.
take 3 (1 : 2 : 3 : 4 : 5 : [])
1 : take 2 (2 : 3 : 4 : 5 : [])
1 : 2 : take 1 (3 : 4 : 5 : [])
-- etc.

至少,一旦你超过go,列表就可以消费了。 eft'不是这种情况 - (++)仍然需要处理,并且这样做与列表长度呈线性关系:

take 3 (eft' 1 5)
take 3 (go 1 5 [])
take 3 (go 2 5 ([] ++ [1]))
take 3 (go 3 5 (([] ++ [1]) ++ [2]))
-- etc.
take 3 ((((([] ++ [1]) ++ [2]) ++ [3]) ++ [4]) ++ [5])
take 3 (((([1] ++ [2]) ++ [3]) ++ [4]) ++ [5])
take 3 ((((1 : ([] ++ [2])) ++ [3]) ++ [4]) ++ [5])
take 3 ((((1 : [2]) ++ [3]) ++ [4]) ++ [5])
take 3 (((1 : ([2] ++ [3])) ++ [4]) ++ [5])
-- etc.
take 3 (1 : ((([2] ++ [3]) ++ [4]) ++ [5]))
1 : take 2 ((([2] ++ [3]) ++ [4]) ++ [5])

情况变得更糟:你必须再次使用列表的剩余尾部为每个元素做到这一点!

1 : take 2 ((([2] ++ [3]) ++ [4]) ++ [5])
1 : take 2 (((2 : ([] ++ [3])) ++ [4]) ++ [5])
1 : take 2 (((2 : [3]) ++ [4]) ++ [5])
1 : take 2 ((2 : ([3] ++ [4])) ++ [5])
-- etc.
1 : take 2 (2 : (([3] ++ [4]) ++ [5]))
1 : 2 : take 1 (([3] ++ [4]) ++ [5])
-- etc.

事实上,take 10掩盖了eft'eft不同的事实是二次方的:

GHCi> last $ eft' 1 10000
10000
(1.83 secs, 4,297,217,200 bytes)
GHCi> last $ eft' 1 20000
20000
(7.59 secs, 17,516,804,952 bytes)
GHCi> last $ eft 1 5000000
5000000
(3.81 secs, 1,080,282,784 bytes)
GHCi> last $ eft 1 10000000
10000000
(7.51 secs, 2,160,279,232 bytes)

为了完整起见,这里是eft''的相应手工评估:

take 3 (eft'' 1 5)
take 3 (1 : eft'' 2 5)
1 : take 2 (eft'' 2 5) -- No need to evaluate `eft'' 2 5` to get the first element.
1 : take 2 (2 : eft'' 3 5)
1 : 2 : take 1 (eft'' 3 5)
-- etc.
1 : 2 : 3 : take 0 (eft'' 4 5) -- No need to go further.
1 : 2 : 3 : []