防止 PHP 函数中堆栈溢出的最佳实践

什么是堆栈溢出？

堆栈溢出是一种运行时错误，当一个函数不断调用自身，在函数调用栈中创建过多的调用时就会发生。这会导致内存不足，进而导致程序崩溃。

避免堆栈溢出的最佳实践

立即学习“PHP免费学习笔记（深入）”；

1. 使用尾递归

尾递归是递归函数的一种特殊形式，其中函数的最后一个操作是调用自身。这可以让编译器优化掉递归调用，避免创建额外的堆栈帧。

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

2. 使用备忘录

备忘录是一种技术，它存储已经完成的函数调用的结果。这可以防止重复计算，从而减少堆栈调用的数量。

$cache = [];

function fibonacci($n) {
    if (!array_key_exists($n, $cache)) {
        if ($n < 2) {
            $cache[$n] = $n;
        } else {
            $cache[$n] = fibonacci($n - 1) + fibonacci($n - 2);
        }
    }

    return $cache[$n];
}

3. 使用迭代

有时，可以使用迭代来重写递归函数，这通常可以避免堆栈溢出。

function factorial_iter($n) {
    $result = 1;
    for ($i = 1; $i <= $n; $i++) {
        $result *= $i;
    }
    return $result;
}

4. 限制函数调用的深度

可以通过设置函数调用的最大深度来限制函数调用的数量。

ini_set('max_execution_time', 30); // 30 秒后执行超时
ini_set('xdebug.max_nesting_level', 100); // 限制嵌套调用深度为 100

实战案例

考虑一个计算阶乘的递归函数：

function factorial($n) {
    if ($n < 2) {
        return $n;
    } else {
        return $n * factorial($n - 1);
    }
}

当传递一个大整数时，此函数可能会导致堆栈溢出。我们可以使用尾递归来优化它：

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

这个优化的函数可以使用尾递归来避免堆栈溢出，即使对于非常大的整数也是如此。