当前位置：首页 > news >正文

LeetCode1137 第N个泰波那契数

news 来源：原创 2025/6/10 2:37:10

泰波那契数列求解：从递归到迭代的优化之路

在算法的世界里，数列问题常常是我们锻炼思维、提升编程能力的重要途径。今天，让我们一同深入探讨泰波那契数列这一有趣的话题。

泰波那契数列的定义

泰波那契序列 Tn 有着独特的定义方式：T0 = 0，T1 = 1，T2 = 1，并且在 n >= 0 的条件下，Tn+3 = Tn + Tn+1 + Tn+2 。简单来说，从第三项之后的每一项，都是它前面三项的和。这与我们熟悉的斐波那契数列有些相似，但又多了一项的参与，也因此带来了一些不同的挑战。

求解任务

给定一个整数 n，我们的目标是返回第 n 个泰波那契数 Tn 的值。这看似简单的任务，却能通过多种不同的算法思路来实现，每种思路都有着其独特的优缺点。

递归法：直观但低效

递归法是最直接的思路，它完全依照泰波那契数列的定义来编写代码。

class Solution {
    public int tribonacci(int n) {
        if (n == 0) {
            return 0;
        }
        if (n == 1 || n == 2) {
            return 1;
        }
        return tribonacci(n - 1) + tribonacci(n - 2) + tribonacci(n - 3);
    }
}

这段代码中，我们通过不断地递归调用自身，计算出每一项的值。当 n 为 0 时，直接返回 0；当 n 为 1 或 2 时，返回 1；否则，返回前三项的和。

然而，这种方法存在着严重的效率问题。因为在递归过程中，会有大量的重复计算。例如，计算 T(n) 时，T(n - 1)、T(n - 2) 和 T(n - 3) 都会被多次计算，随着 n 的增大，计算量会呈指数级增长。

复杂度分析

时间复杂度： $O(3^{n})$ ，由于每次递归调用会产生三个子问题，所以整体时间复杂度为指数级。
空间复杂度： $O(n)$ ，这是因为递归调用栈的深度最大为 n。

记忆化搜索法：优化递归

为了解决递归法中的重复计算问题，记忆化搜索法应运而生。它的核心思想是使用一个数组来记录已经计算过的泰波那契数，当再次需要计算相同的数时，直接从数组中读取，而不需要重复计算。

class Solution {
    int[] memo;
    public int tribonacci(int n) {
        memo = new int[n + 1];
        return helper(n);
    }

    private int helper(int n) {
        if (n == 0) {
            return 0;
        }
        if (n == 1 || n == 2) {
            return 1;
        }
        if (memo[n] != 0) {
            return memo[n];
        }
        memo[n] = helper(n - 1) + helper(n - 2) + helper(n - 3);
        return memo[n];
    }
}

在这段代码中，我们首先创建了一个 memo 数组，用于存储已经计算过的泰波那契数。在 helper 方法中，每次计算前先检查 memo 数组中是否已经存在该值，如果存在则直接返回；否则，计算并将结果存入 memo 数组。

复杂度分析

时间复杂度： $O(n)$ ，因为每个泰波那契数只需要计算一次，后续可以直接从数组中获取。
空间复杂度： $O(n)$ ，主要用于存储 memo 数组。

迭代法：高效且简洁

迭代法是解决泰波那契数列问题的最优方案之一。它通过循环，从已知的初始值开始，逐步计算出后续的每一项。

class Solution {
    public int tribonacci(int n) {
        if (n == 0) {
            return 0;
        }
        if (n == 1 || n == 2) {
            return 1;
        }
        int t0 = 0, t1 = 1, t2 = 1;
        int result = 0;
        for (int i = 3; i <= n; i++) {
            result = t0 + t1 + t2;
            t0 = t1;
            t1 = t2;
            t2 = result;
        }
        return result;
    }
}

在这个实现中，我们利用三个变量 t0、t1 和 t2 来存储当前的前三项，通过循环不断更新这三个变量的值，从而计算出下一项。这种方法避免了递归调用带来的额外开销，代码简洁且高效。

复杂度分析

时间复杂度： $O(n)$ ，只需要一次循环遍历，时间复杂度与 n 成正比。
空间复杂度： $O(1)$ ，只使用了常数级的额外空间，因为我们只需要三个变量来存储中间结果。

总结

通过对泰波那契数列求解问题的探讨，我们看到了不同算法思路的魅力与差异。递归法虽然直观，但效率低下；记忆化搜索法通过优化递归，避免了重复计算，提升了效率；而迭代法则以其简洁高效的特点，成为解决此类问题的首选方法。在实际编程中，我们需要根据具体问题的特点和需求，选择最合适的算法，以实现高效且优质的代码。希望今天的分享能让你对泰波那契数列以及算法优化有更深入的理解，在未来的编程之路上能够更加得心应手。