Haskell中索引的类型签名

Question

欧拉计划第 7 题：第 10 001 个素数是多少？

这是一个接受单个参数 (10001) 并返回第 10001 个素数的函数。 GHCi 没有给我任何问题：

p007nthPrime x = primes !! (x - 1)
    where
        primes :: [Integer]
        primes = sieve [2..]
            where
                sieve (p:xs) = p : sieve [x|x <- xs, x `mod` p > 0]

现在，作为一名优秀的 Haskeller，我想做一个类型签名。但是，将p007nthPrime :: (Integral a) => a -> a 放在顶部会引发此错误：

projecteuler.hs:81:29:
    Couldn't match type `Integer' with `Int'
    Expected type: Int
      Actual type: a
    In the first argument of `(-)', namely `x'
    In the second argument of `(!!)', namely `(x - 1)'
    In the expression: primes !! (x - 1)
Failed, modules loaded: none.

并插入p007nthPrime :: (Num a) => a -> a 做同样的事情。这个函数的正确类型签名是什么？

Answer 1

(!!) 列表索引运算符只需要Ints，而您的primes 列表包含Integers，因此您的签名需要是

p007nthPrime :: Int -> Integer

Int 和 Integer 都是 Integral 类的类型，但一个类型不能同时是 Int 和 Integer。

如果您需要参数也是Integer，我建议使用fromIntegral 函数。

否则，要找出赋予函数/值的类型签名，一个好主意是在 GHCi 中使用 :t 命令：

*Main> :t p007nthPrime
p007nthPrime :: Int -> Integer