面试高频手撕题 | 21.实现一个数组扁平化

一、知识点

扁平化数组意味着将多维数组转换为一维数组，从而减少数组的嵌套层级。扁平化后的数组更易于处理和操作。

扁平化数组的主要知识点包括：

1. 递归：通过递归遍历数组，处理嵌套的子数组。
2. 合并操作：将嵌套的子数组元素与父数组的元素合并在一起。
3. 遍历和扁平化：遍历数组中的每个元素，判断是否为数组，如果是，则进行递归扁平化处理。

二、思路分析

实现数组扁平化的思路如下：

1. 定义一个扁平化函数，接受一个数组作为参数。
2. 使用递归遍历数组。
3. 如果当前元素是一个数组，则递归调用扁平化函数处理该子数组。
4. 将子数组扁平化后的结果与当前父数组的元素合并。
5. 重复步骤 2 至步骤 4，直到处理完整个数组。
6. 返回扁平化后的一维数组。

三、JavaScript 解答

以下是使用 JavaScript 实现数组扁平化的代码示例：

function flattenArray(arr) {
  let flattenedArray = [];
  for (let element of arr) {
    if (Array.isArray(element)) {
      flattenedArray.push(...flattenArray(element));
    } else {
      flattenedArray.push(element);
    }
  }
  return flattenedArray;
}

// 使用示例
let nestedArray = [1, [2, [3, 4], 5], 6];
console.log(flattenArray(nestedArray)); 

四、Java 解答

在 Java 中，你可以使用递归来实现数组扁平化。以下是一个示例代码：

public class ArrayFlattening {
 public static void main(String[] args) {
 int[][] nestedArray = {
 {101, 201},
 {301, 401},
 {501, 601}
 };

 int[] flattenedArray = flattenArray(nestedArray);

 for (int element : flattenedArray) {
 System.out.print(element + " ");
 }
 }

 public static int[] flattenArray(int[][] arr) {
 int[] flattenedArray = new int[arr.length * arr[0].length];
 int index = 0;

 for (int[] row : arr) {
 for (int element : row) {
 flattenedArray[index++] = element;
 }
 }

 return flattenedArray;
 }
}

五、总结

通过上述的一、知识点，二、思路分析，三、JavaScript 解答，四、Java 解答等方面，我们详细讨论了实现数组扁平化的方法和思路。无论是在 JavaScript 还是 Java 中，我们都可以使用递归的方式来遍历数组，将嵌套的子数组元素与父数组的元素合并，从而得到扁平化后的结果。