主页 / coding / 问题 / 77250658
Accepted
Alejandro Caro
Asked: 2023-10-08 00:34:07 +0800 CST

在列表理解格式函数中定义变量

  • 772

我有这个伪代码，可以计算用户给出的数字的阶乘。

Process without_title
     Define i, tmp, x As Integers;
     tmp<-1;
     Write "Enter a number to calculate its factorial";
     Read x;
     For i <- 1 Until x Do
         tmp<-tmp * i;
     EndTo
     Write tmp;
EndProcess

现在我想使用列表理解在 Haskell 中实现相同的代码

import Data.IORef

factorial::Int -> Int
factorial x = [tmp|i<-[1..x], modifyIORef tmp (*i)]

抛出错误说tmp变量未定义

如何定义tmp变量以便我可以使用它？

list

2 个回答

  1. Best Answer

    要使用像这样的可变变量实现阶乘函数，IORef您需要在 IO monad 中工作并使用一个newIORef操作来创建您的IORef，然后使用forM_、traverse_、 或sequence_或类似的函数来对该 IO 操作集进行排序IORef。它可能看起来像：

    import Control.Monad
import Data.IORef

factorial :: Int -> IO Int
factorial x = do
  tmp <- newIORef 1
  forM_ [1..x] $ \i -> modifyIORef' tmp (*i)
  readIORef tmp

    或者，如果您更喜欢使用列表理解，因此它看起来更像您上面的尝试，例如：

    import Data.IORef

factorialIO :: Int -> IO Int
factorialIO x = do
  tmp <- newIORef 1
  sequence_ [modifyIORef' tmp (*i) | i <- [1..x]]
  readIORef tmp

    （使用 strictmodifyIORef'可以提供比lazy 更好的性能modifyIORef。）

    然后可以这样调用：

    main :: IO ()
main = do
  result <- factorial 10
  print $ result

    这通常会比更直接、“纯粹”的解决方案慢，例如：

    factorial :: Int -> Int
factorial x = product [1..x]

    我的基准测试表明IORef基于 的版本比纯基于的版本慢大约两倍product。

    • 3

  2. 嗯，使用变异变量是非常不符合哈斯克尔风格的。这是您需要使用IO Refs 的原因之一：因为 ref 本身不会更改，并且修改其值被视为副作用。

    因此，如果您想手动实现阶乘，则可以使用递归来重复某个任务，并且通过将不同的值传递给参数可以视为使用修改后的值执行此迭代，因此：

    factorial::Int -> Int
factorial 0 = 1
factorial n = n * (factorial (n-1))

    但由于 Haskell 有一些更高级别的实用函数，我们可以生成一个项目列表，然后确定这些项目的乘积。

    • 0

相关问题