第 N 个泰波那契数---序列DP

题目描述

泰波那契序列 T(n) 定义如下：

T(0) = 0, T(1) = 1, T(2) = 1, 且在 n >= 0 的条件下 T(n+3) = T(n) + T(n+1) + T(n+2)

给你整数 n，请返回第 n 个泰波那契数 T(n )的值。

示例 1：

输入：n = 4

输出：4

解释：

T_3 = 0 + 1 + 1 = 2

T_4 = 1 + 1 + 2 = 4

思路

代码

#include <iostream>

using namespace std;

class Solution {
public:
    int tribonacci(int n) {
        if (n == 0) {
            return 0;
        }
        if (n <= 2) {
            return 1;
        }
        int p = 0, q = 0, r = 1, s = 1;
        for (int i = 3; i <= n; ++i) {
            p = q;
            q = r;
            r = s;
            s = p + q + r;
        }
        return s;
    }
};

int main() {
    int n;
    cin >> n;
    Solution a;
    cout << a.tribonacci(n);
}

复杂度分析

• 时间复杂度：O(n)。

• 空间复杂度：O(1)。