js实现flatmap

实现 flatMap 的方法

在 JavaScript 中，flatMap 是一种结合了 map 和 flat 功能的方法，它先对数组的每个元素执行映射函数，然后将结果扁平化一层。以下是几种实现方式：

使用 Array.prototype.flatMap（ES2019 原生支持）

现代浏览器和 Node.js 环境原生支持 flatMap，可以直接调用：

const arr = [1, 2, [3, 4]];
const result = arr.flatMap(x => x);
console.log(result); // [1, 2, 3, 4]

手动实现 flatMap

如果环境不支持原生 flatMap，可以通过以下方式实现：

function flatMap(arr, callback) {
  return arr.reduce((acc, curr) => {
    const mapped = callback(curr);
    return acc.concat(Array.isArray(mapped) ? mapped : [mapped]);
  }, []);
}

// 示例用法
const arr = [1, 2, 3];
const result = flatMap(arr, x => [x * 2]);
console.log(result); // [2, 4, 6]

使用 map 和 concat

另一种实现方式是结合 map 和 concat：

function flatMap(arr, callback) {
  return [].concat(...arr.map(callback));
}

// 示例用法
const arr = [1, 2, 3];
const result = flatMap(arr, x => [x, x * 2]);
console.log(result); // [1, 2, 2, 4, 3, 6]

注意事项

• 原生 flatMap 只能扁平化一层，深层嵌套需要额外处理。
• 手动实现时需确保回调函数的返回值是数组，否则可能导致意外结果。
• 性能敏感场景建议测试不同实现的效率。