2

我无法弄清楚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

4

1 回答 1

6
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
于 2013-06-07T22:27:22.877 回答