如何高效地对任意数量的同长度数组进行逐元素相乘

心靈之曲 2026-01-04 00:00:00 次阅读

本文介绍使用 javascript 的 `reduce` 方法，结合自定义乘法函数，实现对 n 个同长度一维数组的逐元素（按索引）乘积运算，并提供健壮、可复用的解决方案。

在实际开发中，我们常需对多个结构一致的数组（如时间序列数据、特征向量、矩阵行向量等）执行逐元素运算。当数组数量不固定（0 到 n 个）时，硬编码两两相乘不再适用，而 Array.prototype.reduce() 正是处理此类“累积归约”问题的理想工具。

核心思路是：将输入数组列表视为待归约的集合，初始值设为 null，首次迭代时直接取第一个数组作为累加器（acc），后续每次用自定义函数 multiplyArray(acc, curr) 将当前数组 curr 与累加器逐元素相乘，最终得到结果数组。

以下是优化后的完整实现（含边界处理与函数式风格）：

const multiplyArrays = (arrays) => {
  // 空输入：返回空数组
  if (!Array.isArray(arrays) || arrays.length === 0) return [];

  // 单数组：直接返回副本（避免修改原数组）
  if (arrays.length === 1) return [...arrays[0]];

  // 验证所有数组长度一致
  const len = arrays[0].length;
  for (let i = 1; i < arrays.length; i++) {
    if (!Array.isArray(arrays[i]) || arrays[i].length !== len) {
      throw new Error(`All arrays must be of the same length (${len}), but array[${i}] has length ${arrays[i].length}`);
    }
  }

  // 逐元素相乘辅助函数（纯函数，不修改原数组）
  const multiplyTwo = (a, b) => a.map((val, i) => val * b[i]);

  // 使用 reduce 累积相乘
  return arrays.reduce((acc, curr) => multiplyTwo(acc, curr));
};

// 示例调用
const input = [
  [1, 2, 3, 4],
  [1, 2, 3, 4],
  [1, 2, 3, 4],
  [1, 2, 3, 4]
];
console.log(multiplyArrays(input)); // [1, 16, 81, 256]

// 其他测试用例
console.log(multiplyArrays([]));           // []
console.log(multiplyArrays([[5, 6]]));     // [5, 6]
console.log(multiplyArrays([[2, 3], [4, 5]])); // [8, 15]