針對Haskell中的列表的點積產品

Question

我試圖在Prelude庫中僅使用函數的兩個數字列表之間實現一個點積。我寫了下面的功能：

dot :: Num a => [a] -> [a] -> a 
dot x y = sum $ zipWith (*) x y 

我測試如下：

main :: IO() 
main = do 
    let n = 10^6 
     x = (replicate n 2.0) :: [Double] 
     y = (replicate n 3.0) :: [Double] 
    print $ dot x y 
    return() 

不幸的是，這個代碼導致堆棧溢出的空間用於名單100萬件（GHC使用7.6。 3和優化標誌-O2）。

對於這樣一個簡單的例子，我希望ghc能夠執行必要的優化以避免遞歸調用的代價。我錯過了什麼嗎？我的執行錯誤？

Answer 1

sum（曾經是）與foldr一起實施。這對大多數實例Num有點愚蠢的選擇;嚴格的左邊摺疊更好。使用

import Data.List 

sum' :: Num a => [a] -> a 
sum' = foldl' (+) 0 

而不是你的堆棧溢出將消失。