生成前n个素数(Haskell) [英] Generate First n Primes (Haskell)

问题描述

我有这个功能：

generatePrimes :: Integral a => a -> [a]
generatePrimes n = [i | i <- [2..], isPrime i]

我正在尝试获取第一个n素数。我知道我可以通过使用take函数调用main中的函数(并获取列表的第一个n元素)，但我希望能够在函数达到n素数时停止该函数(在函数内部)，以便在main中调用它时，例如：

generatePrimes 8

它将显示仅包含前8个素数的列表。

推荐答案

您所提问题的答案是将take移到generatePrimes的定义中，这样：

generatePrimes :: Integral a => Int -> [a]
generatePrimes n = take n [i | i <- [2..], isPrime i]

如果您必须保持与问题中完全相同的类型签名，则可以使用Data.List中提供的更多态版本的take：

import Data.List
generatePrimes :: Integral a => a -> [a]
generatePrimes n = genericTake n [i | i <- [2..], isPrime i]

(实际上，此实现具有更一般的类型generatePrimes :: (Integral a, Integral i) => i -> [a]。)

但是，这是anti-modular in the presence of lazy evaluation；您应该将列表的使用量控制权交给消费者，而不是生产者。

这篇关于生成前n个素数(Haskell)的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！