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

我想使用递归来完成这个操作，但我卡住了。我试图不使用临时数组来完成它，但是事情似乎变得混乱了。我觉得我错过了一些递归的核心原则。

我意识到我需要一个基本情况。我能想到的唯一情况是当flatten中的当前数组没有子数组时。但是由于我总是通过引用传递数组，所以我没有任何东西可以返回。

我的伪代码是

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

- Nate

---

http://jsfiddle.net/50dnu97z/ - zerkms

ES6: babeljs.io/repl - zerkms

@zerkms - ES6 是一件非常美妙的事情，不是吗 :p - Jaromanda X

我总是通过引用传递数组。当你实现递归时，应该忘记引用：你只接受和返回值。 - zerkms

@Nate 这取决于所使用的编程语言和数据结构。就JS而言，递归解决方案通常效率较低。 - zerkms

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));

- Guffa

---

Array.prototype.splice.apply(arr, [i, 1].concat(arr[i])); 很棒！ - sp00m

Answer 2

这里是一个纯递归版本，完全在原地操作，不构建任何中间数组。

// Deep-flatten an array starting at index `i`.
function flatten(a, i = 0) {                                                                              

  if (i >= a.length)                                                                                      
    // Null case--we are off the end of the array.                                                        
    return;                                                                                               

  var elt = a[i];                                                                                         

  if (!Array.isArray(elt))                                                                                
    // Element is not an array. Proceed to next element.                                                  
    i++;                                                                                                  

  else                                                                                                    
    // Element is an array.                                                                               
    // Unroll non-empty arrays; delete empty arrays.                                                      

    if (elt.length) {                                                                                     
      // Non-empty array.                                                                                 
      // Unroll it into head and tail.                                                                    

      // Shift elements starting at `i` towards end of array.
      // Have to do this recursively too--no loops!                                             
      (function shift(a, i, n = a.length) {                                                               
        if (n > i ) a[n] = a[n-1], shift(a, i, --n);                                                      
      }(a, i));                                                                                           

      // Replace elt with its head, and place tail in slot we opened up.                                  
      a[i] = elt.shift();                                                                                 
      a[i + 1] = elt;                                                                                     
    }                                                                                                     

  else                                                                                                    
    // Array is empty.                                                                                    
    // Delete the element and move remaining ones toward beginning of array. 
    // Again, have to do this recursively!                             
    (function unshift(a, i) {                                                                             
      if (i < a.length) a[i] = a[i+1], unshift(a, ++i);                                                   
      else a.length--;                                                                                    
    }(a, i));                                                                                             

  flatten(a, i);                                                                                          

}                                                                                                         

var arr = [[[2, 3], 4], 5, [6, 7]];                                                                   
flatten(arr);                                                                                             
console.log(arr);                                                                                         

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

这个解决方案使用了ES6的一些特性，例如默认函数参数和Array.isArray。如果您没有ES6，请相应地进行调整。

- user663031

---

修改数据的函数使用起来不方便：你无法将它们组合在一起。 - zerkms

实际上，递归和原地算法之间存在一定的矛盾。对于原地算法而言，递归在某种程度上仅限于取代循环，而不是将其作为组合部分结果的策略。 - user663031

实际上这与递归无关。任何不纯的函数都是不可组合的。 - zerkms