📜  使用 forEach 循环在Java中展平数组流

📅  最后修改于: 2022-05-13 01:54:51.332000             🧑  作者: Mango

使用 forEach 循环在Java中展平数组流

给定Java中的数组流,任务是使用 forEach() 方法展平流。

例子:

Input: arr[][] = {{ 1, 2 }, { 3, 4, 5, 6 }, { 7, 8, 9 }}
Output: [1, 2, 3, 4, 5, 6, 7, 8, 9]

Input: arr[][] = {{'G', 'e', 'e', 'k', 's'}, {'F', 'o', 'r'}}
Output: [G, e, e, k, s, F, o, r]

方法:

  1. 以二维数组的形式获取数组。
  2. 创建一个空列表来收集展平的元素。
  3. 在 forEach 循环的帮助下,将数组的每个元素转换为流并将其添加到列表中
  4. 现在使用 stream() 方法将此列表转换为流。
  5. 现在通过使用 toArray() 方法将流转换为数组来展平流。

下面是上述方法的实现:

示例 1:使用整数数组。

// Java program to flatten a stream of arrays
// using forEach() method
  
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Stream;
  
class GFG {
  
    // Function to flatten a Stream of Arrays
    public static  Stream flattenStream(T[][] arrays)
    {
  
        // Create an empty list to collect the stream
        List list = new ArrayList<>();
  
        // Using forEach loop
        // convert the array into stream
        // and add the stream into list
        for (T[] array : arrays) {
            Arrays.stream(array)
                .forEach(list::add);
        }
  
        // Convert the list into Stream and return it
        return list.stream();
    }
  
    public static void main(String[] args)
    {
  
        // Get the arrays to be flattened.
        Integer[][] arr = {
            { 1, 2 },
            { 3, 4, 5, 6 },
            { 7, 8, 9 }
        };
  
        // Flatten the Stream
        Integer[] flatArray = flattenStream(arr)
                                  .toArray(Integer[] ::new);
  
        // Print the flattened array
        System.out.println(Arrays.toString(flatArray));
    }
}
输出:
[1, 2, 3, 4, 5, 6, 7, 8, 9]

示例 2:使用字符数组。

// Java program to flatten a stream of arrays
// using forEach() method
  
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Stream;
  
class GFG {
  
    // Function to flatten a Stream of Arrays
    public static  Stream flattenStream(T[][] arrays)
    {
  
        // Create an empty list to collect the stream
        List list = new ArrayList<>();
  
        // Using forEach loop
        // convert the array into stream
        // and add the stream into list
        for (T[] array : arrays) {
            Arrays.stream(array)
                .forEach(list::add);
        }
  
        // Convert the list into Stream and return it
        return list.stream();
    }
  
    public static void main(String[] args)
    {
  
        // Get the arrays to be flattened.
        Character[][] arr = {
            { 'G', 'e', 'e', 'k', 's' },
            { 'F', 'o', 'r' },
            { 'G', 'e', 'e', 'k', 's' }
        };
  
        // Flatten the Stream
        Character[] flatArray = flattenStream(arr)
                                    .toArray(Character[] ::new);
  
        // Print the flattened array
        System.out.println(Arrays.toString(flatArray));
    }
}
输出:
[G, e, e, k, s, F, o, r, G, e, e, k, s]