• 如果您觉得本站非常有看点,那么赶紧使用Ctrl+D 收藏吧

java 使用 lamba 表达式处理二维数组

编写数学优化算法时遇到一个问题:如何将一个二维数组满足一定条件的行,筛选出来并组成一个新的数组。

传统思路:

编写一个专门的方法,这个方法返回满足条件的新数组,但数组必须首选确定满足条件的行数,有点麻烦。

或者使用 ArrayList, 使用 add 来添加满足条件的行,但最后还得把 ArrayList 转化成数组,也有点麻烦。

新思路:

使用 lambda 表达式,将二维数组转化成 stream, 使用 filter 过滤条件, 然后将过滤后的数组转化为二维数组。

举例:从一个二维数组中筛选出第四个元素大于5的行。

import java.util.Arrays;


public class HelloWorld {
	
	
	public static void main(String[] args) {
			int[][] a = {
  {3, 6, 8, 10}, {4, 5, 6, 7}, {1, 2, 3, 4}}; // 筛选出第四个元素大于5的行		
						
			int[][] b = Arrays.stream(a).filter(p -> p[3] > 5).map(i -> Arrays.stream(i).toArray())
									.toArray(int[][] :: new);
			
			System.out.println(Arrays.deepToString(b));
	}
}

注: 必须首先用一个 map 将 每一个行引用转化成一维数组,然后在用第二个 toArray 结合 二维数组的构造函数,转化成二维数组。

输出结果:

[[3, 6, 8, 10], [4, 5, 6, 7]]


版权声明:本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系管理员进行删除。
喜欢 (0)