如何在Haskell列表推导中定义tmp变量实现阶乘计算？

我有一段计算用户输入数字阶乘的伪代码：

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变量才能正常使用？

这段代码存在几个核心问题：

• IORef依赖IO上下文：modifyIORef是带副作用的IO操作，只能在IO monad中执行，但你的factorial函数返回纯Int类型，两者无法兼容。
• 列表推导用法错误：列表推导是用于生成纯列表的语法，不能用来执行IO副作用，也无法像命令式循环那样维护可变状态。
• tmp未初始化：代码里完全没有给tmp分配初始值，即便要用IORef，也得先创建并初始化它。

正确实现方式 #

Haskell是纯函数式语言，阶乘最适合用纯函数实现，完全不需要可变状态：

方式1：用内置product函数（最简洁）

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

product函数会计算列表所有元素的乘积，和伪代码里循环累积的逻辑完全一致。

方式2：递归实现（直观体现阶乘定义）

factorial :: Int -> Int
factorial 0 = 1  -- 阶乘定义：0! = 1
factorial n = n * factorial (n - 1)

方式3：模拟命令式可变状态（不推荐，违背Haskell设计思想）

如果一定要复刻伪代码里的可变变量逻辑，必须把代码放在IO上下文里：

import Data.IORef

factorialIO :: Int -> IO Int
factorialIO x = do
  tmp <- newIORef 1  -- 初始化tmp为1
  mapM_ (\i -> modifyIORef tmp (*i)) [1..x]  -- 遍历更新tmp
  readIORef tmp  -- 读取最终结果

main :: IO ()
main = do
  putStrLn "Enter a number to calculate its factorial"
  x <- readLn
  result <- factorialIO x
  print result

完整用户交互代码（推荐） #

结合纯函数实现和用户输入逻辑，完整代码如下：

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

main :: IO ()
main = do
  putStrLn "Enter a number to calculate its factorial"
  x <- readLn
  print $ factorial x

内容的提问来源于stack exchange，提问作者Alejandro Caro