loops - while 循环中的错误

Question

我无法弄清楚我在下面编写的玩具 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 版本:7.6.1

矢量版本:0.10.0.1

Answer 1

test a = while a (\x -> (x!0) < 1) (\x -> U.map (+1) a)

应该是

test a = while a (\x -> (x!0) < 1) (\x -> U.map (+1) x)

或

test a = while a (\x -> (x!0) < 1) (U.map (+1))

否则，您正在迭代的函数将忽略其参数。

---

您的循环增加了一个计数器，但没有使用它。您可以删除它:

while v p f = go v
      where go x = if p x then go (f x) else x