返回素数的IEnumerable:最小的实现

本文关键字：实现 IEnumerable 返回 | 更新日期: 2023-09-27 17:49:48

最近我想知道返回给定数量素数的IEumerable的最小实现是什么。它应该适合这个程序：

static int Main(string[] args)
{
    while(true)
    {
        Console.WriteLine("How many Primes?");
        string line = Console.ReadLine();
        if (line.Trim() == "") break;
        int numPrimes;
        if(!int.TryParse(line.Trim(), out numPrimes)) continue;
        int i = 1;
        foreach(int p in PrimeNumbers(numPrimes))
        {
            Console.WriteLine("{0}: {1}", i++, p);                    
        }
    }
    return 0;
}

返回素数的IEnumerable:最小的实现

我的尝试看起来像这样：

static IEnumerable<int> PrimeNumbers(int numPrimes)
{
    yield return 2; // first prime number
    for(int n=1, p = 3; n < numPrimes; p+=2)
    {
        if (!checkIfPrime(p)) continue;                                
        n++;
        yield return p;              
    }
}
// p > 2, odd
private static bool checkIfPrime(int p)
{
    for (int t = 3; t <= Math.Sqrt(p); t += 2)
    {
        if (p % t == 0) return false;              
    }
    return true;
}

它是一个迭代器，它yield returns所有素数。

最小C#程序的另一个例子：

static IEnumerable<int> PrimeNumbers(int n)
{
    return Enumerable.Range(2, int.MaxValue - 2)
                     .Where(i => ParallelEnumerable.Range(2, Math.Max(0, (int)Math.Sqrt(i) - 1))
                                                   .All(j => i % j != 0))
                     .Take(n);
}

public static IEnumerable<int> PrimeNumbers(int NumberPrimes)
{
    yield return 2;
    for (int i = 3; i < NumberPrimes; i = i + 2)
    {
        bool IsPrime = true;
        System.Threading.Tasks.Parallel.For(2, i, (o, state) =>
        {
            if (i % o == 0)
            {
                IsPrime = false;
                state.Break();
            }
        });
        if (IsPrime)
        {
            yield return i;
        }
    }
}
相关文章：