第五章(5) 数值范围及数值流综合使用案例----获取勾股数勾股数
程序员文章站
2024-01-01 16:09:28
...
1.产生一个数值范围
和数字打交道时,有一个常用的东西就是数值范围。比如,假设你想要生成1和100之间的所有数字。Java 8引入了两个可以用于IntStream和LongStream的静态方法,帮助生成这种范围:range和rangeClosed。这两个方法都是第一个参数接受起始值,第二个参数接受结束值。但range是不包含结束值的,而rangeClosed则包含结束值。让我们来看一个例子:
var numbers = IntStream.rangeClosed(1,100).filter(n->n%2==0);//声明一个数值范围,并筛选偶数
numbers.forEachOrdered(System.out::println);//打印出所有偶数,终端操作
System.out.println(numbers.count());//打印出偶数个数,终端操作,只是展示,实际运行会报错,因为流已经被上一个终端消费掉了
2.数值流案例的集大成者---勾股数流
那么以上案例中flatMap是怎么回事呢?我们看到flatMap中映射成一个流,也就是说flatMap创建了一个流的流,故需要使用使用flatMap扁平化一下,使它成为一个“单一”的流。请注意,你在filter之后调用boxed,从rangeClosed返回的IntStream生成一个Stream<Integer>。这是因为你的map会为流中的每个元素返回一个int数组。而IntStream中的map方法只能为流中的每个元素返回另一个int,这可不是你想要的!你可以用IntStream的mapToObj方法改写它,这个方法会返回一个对象值流
你可以像下面这样采取更好的方案
Stream<double[]> pythagoreanTriples2 = IntStream.rangeClosed(1, 100)
.boxed()
.flatMap
(
a -> IntStream.rangeClosed(a, 100)
.mapToObj( b -> new double[]{a, b, Math.sqrt(a*a + b*b)}).filter(t -> t[2] % 1 == 0)
);
pythagoreanTriples2.limit(6).forEach(t->System.out.println(t[0]+","+t[1]+","+t[2]));