更新:增加递归实现的方法
更新:重构非递归实现的方法


思路

与之前两篇文章( map()的实现 ,filter()的实现 )中的迭代方法不一样,reduce() 是归并方法。

reduce 接收两个参数:

  • 第一个参数是在每一项上调用的函数
  • 该函数接收 4 个参数:
  • 前一个值 prev
    当前值 cur
    项的索引 index
    数组对象 array
  • 第二个可选参数是作为归并基础的初始值

reduce 方法返回一个最终的值

代码表示:

arr.reduce(function(prev, cur, index, arr){}, initialValue)

归并

与之前的迭代不同,归并不是对每一项都执行目标函数,而是可以概括为如下两步:

  • 不断地对数组的前两项“取出”,对其执行目标函数,计算得到的返回值
  • 把上述返回值“填回”数组首部,作为新的 array[0]
  • 持续循环执行这个过程,直到数组中每一项都访问了一次
  • 返回最终结果

举例说明

对数组 [1,2,3] 归并执行 (prev, cur) => prev + cur,流程如图:

[1, 2, 3] // 取出 1 + 2 ,填回 3
[3, 3] // 取出 3 + 3 ,填回 6
[6] // 最终结果为 6

所以得到 6 。

实现

第一版

根据这个思路,得到第一版代码如下

// 第一版
Array.prototype.fakeReduce = function fakeReduce(fn, base) {
  // let arr = base ? this.unshift(base) : this;// 首进,返回新数组的长度,影响原数组 故不能这么写
  let initialArr = this;
  let arr = initialArr.concat(); //得到副本

  if (base) arr.unshift(base); // 当存在归并基础值的参数时,将其从数组首部推入
  let index;

  while (arr.length > 2) {
    index = initialArr.length - arr.length + 1;
    let newValue = fn.call(null, arr[0], arr[1], index, initialArr);
    arr.splice(0, 2); // 删除前两项,影响原数组
    arr.unshift(newValue);// 把 fn(arr[0],arr[1]) 的结果从数组首部推入
  }
  index += 1;
  let result = fn.call(null, arr[0], arr[1], index, initialArr);
  return result;
};

注意点:

  • 队列方法 unshift()
  • 可以从数组首部加入任意个项,
  • 返回值是新数组的长度
  • 影响原数组
  • splice() 方法,高程三将其誉为最强大的数组方法
  • 删除任意数量的项
  • 指定 2 个参数: (删除起始位置, 删除项个数)
  • 插入任意数量的项
  • 指定 3 个参数: (起始位置,0,要插入的项)
  • 第二个参数 0 即为要删除的个数
  • 替换,即删除任意数量的项的同时,插入任意数量的项
  • 指定 3 个参数:(起始位置,要删除的个数, 要插入的任意数量的项)
  • 返回值
  • 始终是一个数组,包含从原始数组中删除的项。
  • 若未删除任何项,返回空数组
  • 影响原数组

改进版

从上面的总结可以看出,splice() 方法完全可以取代 unshift() 方法。

而且,第一版中存在一些重复代码,也可以改进。

由此得到第二版代码

// 第二版
Array.prototype.fakeReduce = function fakeReduce(fn, base) {

  let initialArr = this;
  let arr = initialArr.concat();

  if (base) arr.unshift(base);
  let index, newValue;

  while (arr.length > 1) {
    index = initialArr.length - arr.length + 1;
    newValue = fn.call(null, arr[0], arr[1], index, initialArr);

    arr.splice(0, 2, newValue); // 直接用 splice 实现替换
  }

  return newValue;
};

检测:

let arr = [1, 2, 3, 4, 5];
let sum = arr.fakeReduce((prev, cur, index, arr) => {
  console.log(prev, cur, index, arr);
  return prev * cur;
}, 100);

console.log(sum);

输出:

100 1 0 [ 1, 2, 3, 4, 5 ]
100 2 1 [ 1, 2, 3, 4, 5 ]
200 3 2 [ 1, 2, 3, 4, 5 ]
600 4 3 [ 1, 2, 3, 4, 5 ]
2400 5 4 [ 1, 2, 3, 4, 5 ]
12000

最后加上类型检测等

// 第三版
Array.prototype.fakeReduce = function fakeReduce(fn, base) {
  if (typeof fn !== "function") {
    throw new TypeError("arguments[0] is not a function");
  }
  let initialArr = this;
  let arr = initialArr.concat();

  if (base) arr.unshift(base);
  let index, newValue;

  while (arr.length > 1) {
    index = initialArr.length - arr.length + 1;
    newValue = fn.call(null, arr[0], arr[1], index, initialArr);

    arr.splice(0, 2, newValue); // 直接用 splice 实现替换
  }

  return newValue;
};

递归实现

简易版

const reduceHelper = (f, acc, arr) => {
  if (arr.length === 0) return acc
  const [head, ...tail] = arr
  return reduceHelper(f, f(acc, head), tail)
}

Array.prototype.fakeReduce = function (fn, initialValue) {
  const array = this
  return reduceHelper(fn, initialValue, array)
}

注:acc 即 accumulator, 累计回调的返回值。它是上一次调用回调时返回的累积值或 initialValue。

升级版

支持 cb 函数的全部参数

const reduceHelper = (fn, acc, idx, array) => {
  if (array.length === 0) return acc
  const [head, ...tail] = array
  idx++
  return reduceHelper(fn, fn(acc, head, idx, array), idx, tail)
}

Array.prototype.myReduce = function (cb, initialValue) {
  const array = this
  const [head, ...tail] = array
  const startIndex = initialValue ? -1 : 0

  return initialValue ? reduceHelper(cb, initialValue, startIndex, array) : reduceHelper(cb, head, startIndex, tail)
}

重构非递归

Array.prototype.myReduce = function (cb, initialValue) {
  const array = this
  let acc = initialValue || array[0]
  const startIndex = initialValue ? 0 : 1

  for (let i = startIndex; i < array.length; i++) {
    const cur = array[i]
    acc = cb(acc, cur, i, array)
  }
  return acc
}