我一直在阅读有关JavaScript中的递归函数和尾部调用优化(TCO)的文章。我的目标是克服递归函数中的堆栈溢出:

function factorial(n) {
    function recur(n, acc) {
        if (n === 0) {
            return acc;
        } else {
            return recur(n - 1, n * acc);
        }
    }
    return recur(n, 1);
}

factorial(5); // 120
console.log(factorial(4585759)); // Maximum call stack size exceeded


我发现了如何使用thunktrampoline克服递归函数中的堆栈溢出:

let thunk = function (fn) {
    return function() {
        let args = Array.prototype.slice.apply(arguments);
        return function() { return fn.apply(this, args); };
    };
};

function trampoline(f) {
    while (f && f instanceof Function) {
        f = f();
    }
    return f;
}

function factorial(n) {
    let recur = function(x, n) {
        if (n === 0) {
            return x;
        } else {
            return thunk(recur)(n * x, n - 1);
        }
    };
    return trampoline(thunk(recur)(1, n));
}

console.log(factorial(5)); // 120
console.log(factorial(4585759)); // Infinity


但是,我不喜欢强制编写递归函数的方式。我发现了名为TCO的函数的一些实现:

function tco(fn) {
  var active, nextArgs;
  return function() {
    var result;
    nextArgs = arguments;
    if (!active) {
      active = true;
      while (nextArgs) {
        result = fn.apply(this, [nextArgs, nextArgs = null][0]);
      }
      active = false;
    }
    return result;
  };
}


该函数应允许以下内容:

let factorialToc = tco(function(n) {
    function recur(n, acc) {
        if (n === 0) {
            return acc;
        } else {
            return recur(n - 1, n * acc);
        }
    };
    return recur(n, 1);
});


但它不起作用:

factorialToc(5); // 120
console.log(factorialToc(4585759)); // Maximum call stack size exceeded


是否有任何实用功能来封装TCO?

最佳答案

您将该tco函数应用于非递归factorialToc函数,而不应用于导致堆栈溢出的recur。应该是

function factorialToc(n) {
    const recur = tco(function(n, acc) {
        if (n === 0) {
            return acc;
        } else {
            return recur(n - 1, n * acc);
        }
    });
    return recur(n, 1);
}

10-06 11:19