我无法弄清楚while
我在下面写的玩具循环中的错误是什么。它适用于一个输入,但会挂起其他输入。这是代码 - while 循环接受一个向量、一个向量上的谓词和向量上的转换函数,并返回另一个向量:
import Data.Vector.Unboxed as U
while :: Vector Int -> (Vector Int -> Bool) -> (Vector Int -> Vector Int) -> Vector Int
while v p f = go 0 v
where go n x = if (p x) then go (n+1) (f x)
else x
test :: Vector Int -> Vector Int
test a = while a (\x -> (x!0) < 2) (\x -> U.map (+1) a)
main = print $ test (U.fromList [0])
这挂在执行main
. 另一方面,如果我test
改为:
test a = while a (\x -> (x!0) < 1) (\x -> U.map (+1) a)
我得到一个终止结果(ghci
下面的输出):
λ: main
fromList [1]
我觉得我一定错过了什么。我仔细查看了该功能,但无法弄清楚我做错了什么。看起来谓词不能执行超过两次。
顺便说一句,它适用于其他类型,如Int
.
while :: Int -> (Int -> Bool) -> (Int -> Int) -> Int
while i p f = go 0 i
where go n x = if (p x) then go (n+1) (f x)
else x
test :: Int -> Int
test a = while a (\x -> x < 2) (\x -> x+1)
main = print $ test 0
ghci 输出:
λ: main
2
GHC version: 7.6.1
Vector version: 0.10.0.1