Haskell自定义函数接收字符串输入出错，求解决方法

作为Haskell新手，我尝试编写代码让自定义函数接收字符串输入，并根据main中传入的参数执行后续操作。编写的代码如下：

main :: IO()
main=do
{
    print(f 2)
}
f::Int->Int
f n=do
{
    inp<-getLine;
    if (inp=="hello") then (n+2);
    else                   (n);
}

我把f的类型定义为Int->Int，但运行后出现以下错误：

[1 of 1] Compiling Main             ( main.hs, main.o )
main.hs:9:5: error:
    • Couldn't match expected type ‘Int’ with actual type ‘IO b0’
    • In a stmt of a 'do' block: inp <- getLine
      In the expression:
        do inp <- getLine
           if (inp == "hello") then (n + 2) else (n)
      In an equation for ‘f’:
          f n
            = do inp <- getLine
                 if (inp == "hello") then (n + 2) else (n)
  |
9 |     inp<-getLine;
  |     ^^^^^^^^^^^^
main.hs:10:29: error:
    • Couldn't match expected type ‘IO b0’ with actual type ‘Int’
    • In the expression: (n + 2)
      In a stmt of a 'do' block:
        if (inp == "hello") then (n + 2) else (n)
      In the expression:
        do inp <- getLine
           if (inp == "hello") then (n + 2) else (n)
   |
10 |     if (inp=="hello") then (n+2);
   |                             ^^^
main.hs:11:29: error:
    • Couldn't match expected type ‘IO b0’ with actual type ‘Int’
    • In the expression: (n)
      In a stmt of a 'do' block:
        if (inp == "hello") then (n + 2) else (n)
      In the expression:
        do inp <- getLine
           if (inp == "hello") then (n + 2) else (n)
   |
11 |     else                   (n);
   |                             ^

请问该如何修正这些错误？

问题根源 #

你的代码存在两个核心问题：

1. 类型定义错误：f函数中调用了getLine（属于IO操作，返回IO String类型），因此整个do块的返回值必须是IO包裹的类型，而非直接的Int。你定义的Int->Int与实际返回的IO Int类型冲突，触发第一个错误。
2. do块返回值规则违反：Haskell的do块最后一行必须是一个IO动作，而你直接返回了n+2或n这类纯Int值，不符合do块的语法要求，导致后两个错误。

修正方案 #

1. 修正函数类型声明

将f的类型改为Int -> IO Int，明确它是一个接收Int参数、执行IO操作并返回IO包裹的Int值的函数。

2. 用return包装纯值

在if的两个分支中，用return把纯Int值包装成IO动作，让do块的最后一行符合IO类型要求。这里的return不是“返回”的意思，而是将纯值转换为IO动作的工具函数。

3. 调整main函数的调用逻辑

因为f 2返回的是IO Int，不能直接传给print（print需要纯值），所以要在main的do块中先绑定f 2的结果，再传入print。

修正后的代码 #

main :: IO ()
main = do
  result <- f 2
  print result

f :: Int -> IO Int
f n = do
  inp <- getLine
  if inp == "hello"
    then return (n + 2)
    else return n

额外说明 #

• Haskell严格区分纯函数与IO操作：纯函数不能执行任何IO动作，执行IO的函数必须返回IO类型的值。
• 可以去掉原代码中的大括号和分号，Haskell的缩进语法足够清晰，能让代码更易读。

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