java8求质数的收集器

质数收集器

原理是：对小于自然数的平方根的所有质数求余

/**
 * 质数收集器
 *
 * @author YanZhen
 * @since 2019-04-11 09:59
 */
public class PrimeNumbersCollector implements Collector<Integer, Map<Boolean, List<Integer>>, Map<Boolean, List<Integer>>> {
  @Override
  public Supplier<Map<Boolean, List<Integer>>> supplier() {
    // 初始值
    return () -> new HashMap<Boolean, List<Integer>>() {{
      put(true, new ArrayList<>());
      put(false, new ArrayList<>());
    }};
  }

  @Override
  public BiConsumer<Map<Boolean, List<Integer>>, Integer> accumulator() {
    // 筛选质数和非质数
    return (booleanListMap, integer) -> booleanListMap.get(isPrime(booleanListMap.get(true), integer)).add(integer);
  }

  /**
   * 判断是否为质数
   * @param integers 质数组
   * @param integer 自然数
   * @return 是否质数
   */
  private Boolean isPrime(List<Integer> integers, Integer integer) {
    return takeWhile(integers, i -> i <= (int) Math.sqrt((double) integer)).stream().noneMatch(i -> integer % i == 0);
  }

  @Override
  public BinaryOperator<Map<Boolean, List<Integer>>> combiner() {
    // 该操作不能调用该方法
    // 并行计算时需要拆分数据进行计算，计算质数不能用并行
    return (map1, map2) -> {
      map1.get(true).addAll(map2.get(true));// 质数
      map1.get(false).addAll(map2.get(false));// 非质数
      return map1;
    };
  }

  @Override
  public Function<Map<Boolean, List<Integer>>, Map<Boolean, List<Integer>>> finisher() {
    // 返回值
    return Function.identity();
  }

  @Override
  public Set<Characteristics> characteristics() {
    // 定义操作类型
    return Collections.unmodifiableSet(EnumSet.of(Characteristics.IDENTITY_FINISH));
  }

  /**
   * 筛选符合条件的项目
   * @param list 所有项目
   * @param p 条件
   * @param <A> 类型
   * @return 符合条件的项目
   */
  private static <A> List<A> takeWhile(List<A> list, Predicate<A> p) {
    // 遍历
    for (int i = 0; i < list.size(); i++) {
      // 如果不符合条件
      if (!p.test(list.get(i))) {
        // 返回之前的数据
        return list.subList(0, i);
      }
    }
    // 都符合条件的情况
    return list;
  }
}

 Test

public static void main(String[] args) {
  int n = 10_000;
  final Map<Boolean, List<Integer>> map = IntStream.rangeClosed(2, n).boxed().collect(new PrimeNumbersCollector());
  System.out.println(map.get(true));
}

结果：

[2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, ......, 9887, 9901, 9907, 9923, 9929, 9931, 9941, 9949, 9967, 9973]