Question

我试图创建一个素数的无限IntStream，它必须很快 - 在几秒钟内产生大约一百万个素数。所以我尝试创建用过滤器调用生成的流 IntStream 本身：

IntStream primeStream = IntStream.iterate(2, n -> n + 1)
                .filter(PrimeStream::isPrime);

isPrime ，如下所示：

public boolean isPrime(int n, IntStream primeStream) {
        primeStream.forEach((p) -> {
            if (n % p == 0)
                return false;
            if (p > Math.sqrt(n))
                break;
        });
        return true;
    }

那可能吗？或者还有其他方式使用Stream来完成我的任务吗？

Answer 1

无限IntStream的整个点是它不消耗内存，即它不“记住”流式值。它只生成新的int值。

如果你需要记住价值观，你需要自己做。

我建议使用IntStream.generate(IntSupplier s)，IntSupplier会记住以前生成的素数。

public final class PrimeGenerator implements IntSupplier {

    private final List<Integer> primes = new ArrayList<>();

    @Override
    public int getAsInt() {
        int prime;
        final int count = this.primes.size();
        if (count == 0)
            prime = 2;
        else if (count == 1)
            prime = 3;
        else {
            prime = this.primes.get(count - 1) + 2;
            while (! isPrime(prime))
                prime += 2;
        }
        this.primes.add(prime);
        return prime;
    }

    private boolean isPrime(int n) {
        for (int p : this.primes) {
            if (n % p == 0)
                return false;
            if (p * p > n)
                return true;
        }
        return true;
    }
}

像这样使用：

long start = System.nanoTime();
System.out.println(IntStream.generate(new PrimeGenerator()).limit(1000000).max().getAsInt());
long stop = System.nanoTime();
System.out.println("Elapsed: " + (stop - start) / 1000000000d + " seconds");

将显示第100万个素数：

15485863
Elapsed: 1.749469734 seconds

在其迭代过滤器函数中调用IntStream

1 个答案: