一、介绍
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;
}
};