F#：想要重新初始化枚举器

Question

我有一系列素数除数，我想为每个素数候选者迭代。我使用 GetEnumerator() MoveNext() 和 Current。我无法重新初始化枚举器以从头开始。我尝试了 Reset()，它已编译，但运行时错误未实现。

我正在使用 F# 2.0 Interactive build 4.0.40219.1

有什么建议吗？

问候， 道格

---

为了澄清问题：对于每个素数候选者 N，我想遍历素数除数序列（最多约 sqrt N）并完全分解 N 或确定它是否为素数。使用 GetEnumerator、MoveNext、Current 方法适用于第一个主要候选人，但在第二个主要候选人上，我想从头开始迭代我的除数序列。看来，这样做的唯一方法是创建一个新的迭代器（这对于大量素数候选者来说很尴尬）或创建一个新的素数序列（我不想这样做）。

使用“seqPrimes 中的除数”之类的建议似乎在停止之前用尽了所有除数，但我想在素数除数除以主要候选人时立即停止。

如果上述陈述中我的逻辑有错误，请告诉我。

我调查了 Seq.cache，这对我有用。结果代码如下：

---

// Recursive isprime function (modified from MSDN)
let isPrime n =
    let rec check i =
        i > n/2 || (n % i <> 0 && check (i + 2))
    if n = 2 then true
    elif (n%2) = 0 then false
    else check 3

let seqPrimes = seq { for n in 2 .. 100000 do if isPrime n then yield n }

// Cache the sequence to avoid recomputing the sequence elements.
let cachedSeq = Seq.cache seqPrimes


// find the divisors of n (or determine prime) using the seqEnum enumerator 
let rec testPrime n (seqEnum:System.Collections.Generic.IEnumerator<int>) =
  if n = 1 then printfn "completely factored"
  else
    let nref = ref n
    if seqEnum.MoveNext() then
      let divisor = seqEnum.Current
      //printfn "trial divisor %A" divisor
      if divisor*divisor > n then printfn "prime %A" !nref
      else
        while ((!nref % divisor) = 0) do
          printfn "divisor %A" divisor
          nref := !nref / divisor
        testPrime !nref seqEnum

// test
for x = 1000000 to 1000010 do
  printfn "\ndivisors of %d = " x
  let seqEnum = cachedSeq.GetEnumerator()
  testPrime x seqEnum
  seqEnum.Dispose()   // not needed

Answer 1

如果您的意思是您尝试重置枚举器的原因是重新生成素数序列的高成本，您可以考虑caching 您的序列。这种使用序列的方式对于 F# 来说是惯用的。为了向您展示如何做到这一点，我建议您参考以下来自 this context 的 sn-p：

let rec primes = 
    Seq.cache <| seq { yield 2; yield! Seq.unfold nextPrime 3 }
and nextPrime n =
    if isPrime n then Some(n, n + 2) else nextPrime(n + 2)
and isPrime n =
    if n >= 2 then
        primes 
        |> Seq.tryFind (fun x -> n % x = 0 || x * x > n)
        |> fun x -> x.Value * x.Value > n
    else false

您可以使用这个 sn-p 来查看这里重新枚举的惩罚可以忽略不计。

谈到IEnumerator的Reset()方法，我记得在当前的F#中没有实现它，即抛出System.NotSupportedException。有关理由，请参阅MSDN reference。

补充： 为了使用您在下面建议的测试对其进行测试：

for x in [1000000..1000010] do
    printfn "\ndivisors of %d" x
    primes
    |> Seq.takeWhile ((>) (int(sqrt(float x))))
    |> Seq.iter (fun n -> if x%n = 0 then printf "%d " n)

在我的笔记本电脑上执行测试只需要 3 毫秒。