Java Stream来写算法07——使用一个数据流来查找孪生素数

总目录

在上一篇文章中，使用了两个数据流来对应找出孪生素数，方法感觉有些取巧，所以就在想如何用一个数据流来查找到孪生素数？

• 因为有把一切皆当作对象的角度，所以在解决这个想法时，自然而然地就会想到：孪生素数，本质上就是一个数据对。把这样的数据对作为一个对象，难道不是理所应当的吗？

数对对象

// 数对对象
@Getter@Setter@ToString@AllArgsConstructor// 为了节约空间，注释写到一行里
class TwinValue {
    private int firstValue;
    private int secondeValue;
}

• 有了lombok可以把Java Bean代码写得超级简单，对开源大神们唯有心怀感恩，有机会要表达敬意
• 接下来的问题就是如何生成向后无限延展的数对流了
• 每生成一个数对，就是要new出来一个TwinValue对象，因为写new代码太难看，所以用方法引用来增加美感（本质是炫技）

BiFunction<Integer, Integer, TwinValue> biFunctionTwinOdd = TwinValue::new;

• 上述代码本质就是用两个输入参数，来生成一个TwinValue对象
• 这个代码不报错的基础在于写TwinValue这个类时加入@AllArgsConstructor注解，这个注解的意思就是生成有两个参数的构造器
• 准备工作还有一项，就是筛选器中使用的判断素数的函数式接口predicateTwinIsPrime，这个看完整代码
• 好了，可以用一行代码生成孪生素数对了！

Stream<TwinValue> twinOddStream =
    Stream.iterate(biFunctionTwinOdd.apply(3, 5),
        to -> biFunctionTwinOdd.apply(to.getFirstValue() + 2, to.getSecondeValue() + 2))
                .filter(predicateTwinIsPrime).limit(100);// 把limit代码注释掉，就可以生成无限流

• 代码的意思是，首先数据流的起点是（3，5）这个孪生素数对，接下来生成的就是（5，7），（7，9），（9，11）……这样的奇数数对了
• 使用filter把其中全是素数的筛选出来，结果就是孪生素数对了

完整代码

Predicate<TwinValue> predicateTwinIsPrime = v -> {
    int first = v.getFirstValue();
    int second = v.getSecondeValue();
    if (first == 3 && second == 5) return true;
    if (first % 2 == 0 || second % 2 == 0) return false;
    int sqrtValue = Double.valueOf(Math.sqrt(second)).intValue();
    return !IntStream.iterate(3, ov -> ov + 2).limit(sqrtValue)
            .filter(factor -> factor <= sqrtValue)
            .anyMatch(factor -> first % factor == 0 || second % factor == 0);
};

BiFunction<Integer, Integer, TwinValue> biFunctionTwinOdd = TwinValue::new;

private void getTwinPrimeWithOneStream() {
    Stream<TwinValue> twinOddStream = Stream.iterate(biFunctionTwinOdd.apply(3, 5),
            to -> biFunctionTwinOdd.apply(to.getFirstValue() + 2, to.getSecondeValue() + 2))
            .filter(predicateTwinIsPrime).limit(100);
    Iterator<TwinValue> iterator = twinOddStream.iterator();
    int index = 0;
    while (iterator.hasNext()) {// 将数列内容打印出来
        System.out.printf("%d: %s\n", ++index, iterator.next());
    }
}

以这个思路为基础，就可以生成斐波那契数列了，下一篇再说。