Javascript:使用递归将多维数组就地展平

Question

我有以下代码可以展平多维数组

var x = [[[2, 3], 4], 5, [6, 7]];

function flatten(arr) {

  for (var i = 0; i < arr.length; i++) {
    if (arr[i].constructor === Array) {
      subArr = arr[i];
      // Shift the array down based on the space needed for the sub array
      for (var j = arr.length + subArr.length - 2; j > i + subArr.length - 1; j--) {
        arr[j] = arr[j - subArr.length + 1];
      }
      // Insert sub array elements where they belong
      for (var k = 0; k < subArr.length; k++) {
        arr[i + k] = subArr[k]; 
      }
      // Look at arr[i] again in case the first element in the subarray was another array;
      i--;
    }
  }
}

flatten(x);

JSBin 在这里:http://jsbin.com/gazagemabe/edit?js,console

我想使用递归来做到这一点，但我最终陷入困境。我试图在不保留临时数组的情况下做到这一点，但事情似乎变得不正常了。我觉得我缺少递归的一些核心原则。

我意识到我需要一个基本案例。我能想到的唯一情况是当前处于展平状态的数组没有子数组。但是因为我总是通过引用传递数组，所以我没有什么可返回的。

我的伪代码是

function flatten(arr) 

  loop through arr
    if arr[index] is an array
      increase the array length arr[index] length
      flatten(arr[index])
    else
      // unsure what to do here to modify the original array

Answer 1

从内到外进行递归展平。首先对找到的数组调用 flatten 函数，然后将该数组(现在是扁平的)的内容移动到父数组中。通过数组向后循环，这样您就不必为插入的项目调整循环变量。

如您所知，插入的数组是平面的，您可以使用 splice 方法用其项目替换数组。

它的工作原理是这样的:

start with
[[[2, 3], 4], 5, [6, 7]]

flatten [6,7] (which is already flat) and insert:
[[[2, 3], 4], 5, 6, 7]

flatten [[2, 3], 4] recursively calls flatten [2,3] and inserts in that array:
[[2, 3, 4], 5, 6, 7]
then it inserts [2, 3, 4]:
[2, 3, 4, 5, 6, 7]

代码:

function flatten(arr) {
  for (var i = arr.length - 1; i >= 0; i--) {
    if (arr[i].constructor === Array) {
      flatten(arr[i]);
      Array.prototype.splice.apply(arr, [i, 1].concat(arr[i]));
    }
  }
}

var x = [[[2, 3], 4], 5, [6, 7]];

flatten(x);

// Show result in snippet
document.write(JSON.stringify(x));