一道字节笔试题，实现一个异步求和函数

题目：

提供一个异步 add 方法如下，需要实现一个 await sum(...args) 函数：

function asyncAdd(a, b, callback) {
  setTimeout(function () {
    callback(null, a + b);
  }, 1000);
}

简化：两数之和

我们先来简单的实现一个异步两数之和函数

function sumT(a, b) {
    return await new Promise((resolve, reject) => {
        asyncAdd(a, b, (err, res) => {
            if(!err) {
                resolve(res)
            }
            reject(err)
        })
    })
}

// 测试
const test = await sumT(1, 2)
console.log(test)
// 3

加深：多数之和

上面我们实现了两数之和，然后扩展到多数之和喃？

提到数组求和问题，我们首先想到的是 reduce

reduce() 方法对数组中的每个元素执行一个由您提供的reducer函数(升序执行)，将其结果汇总为单个返回值。

—— MDN

arr.reduce(callback(acc, cur[, idx[, arr]])[, initialValue])

callback 函数接收4个参数:

• acc ：累计器
• cur ：当前值
• idx ：当前索引
• arr ：源数组

其中， initialValue 可选，

• 如果有  initialValue ：acc 取值为 initialValue ， cur 取数组中的第一个值
• 如果没有：acc 取数组中的第一个值， cur 取数组中的第二个值

const arr = [1, 2, 3, 4];
const reducer = (acc, cur) => acc + cur;

// 1 + 2 + 3 + 4
console.log(arr.reduce(reducer));
// 输出: 10

// 5 + 1 + 2 + 3 + 4
console.log(arr.reduce(reducer, 5));
// 输出: 15

关于本题：来自@champkeh

设置初始值为 Promise.resolve(0) ，经历 5 次求和：

function sum(...args) {
    return new Promise(resolve => {
        args.reduce((acc, cur) => acc.then(total => sumT(total, cur)), Promise.resolve(0)).then(resolve)
    })
}

// 测试
await sum(1, 2, 3, 4, 5)
// 15

但这存在一个耗时较长的问题，我们可以计算下时间：

console.time("sum")
// 测试
await sum(1, 2, 3, 4, 5)
// 15
console.timeEnd("sum")

也就是说，我们每次求和都会花费 1s，串行异步求和，这显然不是最优的

优化：使用 Promise.all

我们可以两两一组，使用 Promise.all 求和，再把和两两一组继续求和…..，知道只剩余一个就是最终的结果

async function sum(...args) {
    // 用于考察每次迭代的过程
    console.log(args) 
    
    // 如果仅有一个，直接返回
    if(args.length === 1) return args[0]
    let result = []
    // 两两一组，如果有剩余一个，直接进入
    for(let i = 0; i < args.length - 1; i+=2) {
        result.push(sumT(args[i], args[i + 1]))
    }
    if(args.length%2)  result.push(args[args.length-1])
    // Promise.all 组内求和
    return sum(...await Promise.all(result))
}

// 测试
test = await sum(1, 2, 3, 4, 5)
// 15

console.time("sum")
await sum(1, 2, 3, 4, 5)
console.timeEnd("sum")