LeeCode_1137. 第 N 个泰波那契数(动态规划+递归+矩阵快速幂)

一、介绍

1.题目描述

题目链接：https://leetcode-cn.com/problems/n-th-tribonacci-number/

泰波那契序列 Tn 定义如下： 

T0 = 0, T1 = 1, T2 = 1, 且在 n >= 0 的条件下 Tn+3 = Tn + Tn+1 + Tn+2

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

2.测试样例

4  # 4

25  # 1389537

37  # 2082876103

3  # 2

二、题解

1、动态规划🟢

1、当 n < 3，直接返回对应值。

2、当 n > 3，定义 abc，sum = a + b + c，并更新abc

class Solution {
public:
    int tribonacci(int n) {
        if(n==0) return 0;
        if(n<=2) return 1;
        int a=0,b=1,c=1;
        for(int i=3;i<n;i++){
            int t=a+b+c;
            a=b;b=c;c=t;
        }
        return a+b+c;
    }
};

2、递归🔴

递归公式在题目中就已经给出了

但是直接递归在该题中不可用，因为时间超限。

需要对递归进行一定的调整，加入一个temp[38]数组，记录过程中的数，防止重复计算超时【∵0 ≤ n ≤ 37】

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

// 直接递归
// class Solution {
// public:
//     int tribonacci(int n) {
//         if(n==0) return 0;
//         if(n<=2) return 1;
//         return tribonacci(n-1)+tribonacci(n-2)+tribonacci(n-3);
//     }
// };

 3、矩阵快速幂

对于该题，有如下递推关系

 因此，我们构造矩阵M，进行相应运算可得到结果矩阵的第一个数值为我们所需要的解。

class Solution {
public:
    int tribonacci(int n) {
        if(n==0)return 0;
        if(n<=2)return 1;
        // 矩阵 M
        vector<vector<long>> m={{1,1,1},{1,0,0},{0,1,0}};
        // 求 M 的 n-2 次方,由于 pow 初始为平方，所以传入的参数-1
        vector<vector<long>> temp=pow(m,n-3);
        // T2,T1,T0
        vector<vector<long>> t={{1},{1},{0}};
        // M的n-2次方的矩阵第一行与t矩阵相乘即为解
        int ans=temp[0][0]*t[0][0]+temp[0][1]*t[1][0]+temp[0][2]*t[2][0];
        return ans;
    }
    // 矩阵乘法
    vector<vector<long>> pow(vector<vector<long>>& a,int n){
        // 如果求矩阵的一次方，直接返回矩阵
        if(n==0) return a;
        vector<vector<long>> b=a;
        vector<vector<long>> c(3, vector<long>(3));
        while(n>0){
            // 矩阵相乘
            for(int i=0;i<3;i++){
                for(int j=0;j<3;j++){
                    c[i][j]=b[i][0]*a[0][j]+b[i][1]*a[1][j]+b[i][2]*a[2][j];
                }
            }
            b=c;
            n--;
        }
        return c;
    }
};