javascript - 使用 Javascript 的递归问题

Question

我正在应对 FreeCodeCamp 的这项挑战

Flatten a nested array. You must account for varying levels of nesting.

我正在尝试使用递归来解决它。

这是我的代码:

function steamroller(arr) {
    var flatArray = [];

    function flatten(obj) {
        if (!Array.isArray(obj)) {
            return(obj);
        }
        for (var i in obj) {
            return flatten(obj[i]);
        }
    }

    flatArray.push(flatten(arr));

    console.log(flatArray);
}

steamroller([1, [2], [3, [[4]]]]);

此日志:

Array [ 1 ]

我可以看到问题所在，return 语句破坏了 for 循环，因此仅返回第一个值。

但是，如果我省略返回并仅使用:

flatten(obj[i]);

我回来了:

Array [ undefined ]

我应该怎么做才能解决这个问题？

Answer 1

您需要将项目添加到结果数组中。

function flatten(arr){
  var flat = [];

  arr.forEach(function(item){     
    flat.push.apply(flat, Array.isArray(item) ? flatten(item) : [item]);      
  });

  return flat;
}