使用rest参数将参数展平为一维数组

时间:2016-07-14 11:00:47

标签: javascript arrays ecmascript-6 reduce flatten

在尝试将多维数组展平为一维数组时,我开发了以下代码(http://codepen.io/PiotrBerebecki/pen/KrkdPj)。

例如,

flattenArray([1, [2, 3], 4, [5, [6]], 7])
// should return [1, 2, 3, 4, 5, 6, 7]


// Flatten an array
function flattenArray(input) {
  return input.reduce(function(prev, curr) {
    if (Array.isArray(curr)) {return prev.concat(flattenArray(curr));}
    else {return prev.concat(curr);}
  }, []);
}

console.log(  flattenArray([1, [2, 3], 4, [5, [6]], 7])  );
// [1, 2, 3, 4, 5, 6, 7]

以上似乎可行。

现在,我已经被要求制作另一个函数,而不是数组将为数字和数组的混合实现相同的结果。

例如,

flattenMix(1, [2, 3], 4, [5, [6]], 7);
// should return [1, 2, 3, 4, 5, 6, 7]

我通过添加rest参数稍微修改了上述内容,以便该函数可以接受任意数量的参数。但是我收到maximum call stack错误。你知道下面的功能有什么问题吗?

// Flatten arguments (arbitrary length) consisting of a mix of numbers and arrays
function flattenMix(...input) {
  return input.reduce(function(prev, curr) {
    if (Array.isArray(curr)) {return prev.concat(flattenMix(curr));}
    else {return prev.concat(curr);}
  }, []);
}

console.log(  flattenMix(1, [2, 3], 4, [5, [6]], 7)  ); // should return [1, 2, 3, 4, 5, 6, 7]

更新1

下面的答案和建议表明我应该在递归调用...中使用flattenMix(...curr)扩展运算符。这样,当curr是一个数组(经过测试)时,curr的包含元素将传递给flattenMix函数(而不是curr数组)。

4 个答案:

答案 0 :(得分:2)

嗯,很容易。你的第一个方法已经可以处理一个数组,因为arguments基本上是一个数组,我们可以添加一个额外的行。

// Flatten an array
function flattenArray(input) {
  return input.reduce(function(prev, curr) {
    if (Array.isArray(curr)) {
      return prev.concat(flattenArray(curr));
    } else {
      return prev.concat(curr);
    }
  }, []);
}

function flattenMix() {
  return flattenArray([].slice.call(arguments));
}

console.log(flattenMix(1, [2, 3], 4, [5, [6]], 7));

答案 1 :(得分:2)

好吧,你可以使用点差运算符。



function flattenMix(...input) {
    return input.reduce(function (prev, curr) {
        if (Array.isArray(curr)) {
            return prev.concat(flattenMix(...curr));
            //                            ^^^^^^^
        } else {
            return prev.concat(curr);
        }
    }, []);
}

console.log(flattenMix(1, [2, 3], 4, [5, [6]], 7));




答案 2 :(得分:1)

您的问题是您在递归而不是数组项中发送数组作为第一个参数。 您可以使用flatten.apply(null, arr)flatten(...arr)

// Flatten arguments (arbitrary length) consisting of a mix of numbers and arrays

function flatten(...input) {
  return input.reduce(function(prev, curr) {
    return prev.concat(
      Array.isArray(curr) ? flatten(...curr) : curr
    );
  }, []);
}

console.log(flatten(1, [2, 3], 4, [5, [6]], 7)); // should return [1, 2, 3, 4, 5, 6, 7]

这样做有趣吗

如果将数组数组转换为字符串,则会为您展平数组,您应该做的唯一事情就是将字符串数转换为Number

function flattenArgs() {
  return Array.from(arguments).toString()
    .split(',').map(Number);
}

console.log(flattenArgs(1, [2, 3], 4, [5, [6]], 7));

答案 3 :(得分:0)

有一个新方法Array.flat()

var arr1 = [1, 2, [3, 4]];
arr1.flat(); 
// [1, 2, 3, 4]

var arr2 = [1, 2, [3, 4, [5, 6]]];
arr2.flat();
// [1, 2, 3, 4, [5, 6]]

var arr3 = [1, 2, [3, 4, [5, 6]]];
arr3.flat(2);
// [1, 2, 3, 4, 5, 6]

more info