LeetCode 1137 | 第N个泰波那契数

LeetCode 1137 | 第N个泰波那契数

题目

题解

与斐波那契数列相似，解法也都相同

1. 递归：由于测试数据较大，该题使用递归会超时

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);
    }
};

2. 动态规划：简单的递推

class Solution {
public:
    int tribonacci(int n) {
        int arr[38];
        arr[0]=0;
        arr[1]=1;
        arr[2]=1;
        for(int i=3;i<38;i++){
            arr[i]=arr[i-1]+arr[i-2]+arr[i-3];
        }
        return arr[n];
    }
};

3. 滚动数组：dp的简化，仅使用常量空间

class Solution {
public:
    int tribonacci(int n) {
        if(n==0)return 0;
        int p1=0,p2=1,p3=1;
        for(int i=3;i<=n;i++){
            int tmp=p1+p2+p3;
            p1=p2;
            p2=p3;
            p3=tmp;
        }
        return p3;
    }
};