leetCode-第 N 个泰波那契数
第N个泰波那契数
题干
第 N 个泰波那契数
泰波那契序列 Tn 定义如下:
T0 = 0, T1 = 1, T2 = 1, 且在 n >= 0 的条件下 Tn+3 = Tn + Tn+1 + Tn+2
给你整数 n,请返回第 n 个泰波那契数 Tn 的值。
提示:
0 <= n <= 37
答案保证是一个 32 位整数,即 answer <= 2^31 - 1。
#思路
思路
- 方法一:动态规划
泰波那契数的边界条件是 T(0)=0, T(1)=1, T(2)=1T(0)=0,T(1)=1,T(2)=1。
当 n>2n>2 时,每一项的和都等于前三项的和,因此有如下递推关系:T(n)=T(n-1)+T(n-2)+T(n-3)
设置一个大小为4的数组。
通过不断赋值实现最后一个数一直是前三个数之和。
时间复杂度为O(n)
空间复杂度为O(1)
- 方法二:矩阵快速幂
所以只要能通过矩阵乘法,求出矩阵快速幂既可以把Tn快速求出。
时间复杂度为O(log n)
空间复杂度为O(1)
实现代码
class Solution {
public int tribonacci(int n){//矩阵快速乘,快速幂应用
if(n==0){
return 0;
}
if(n<=2){
return 1;
}
int [][] a={{1,1,1},{1,0,0},{0,1,0}};//快速乘关系矩阵
int [][] res=pow(a,n);
return res[0][2];//根据递推矩阵res[0][2]是T(n)的值 存放的是T(N)的值 res[0][0]存放的是T(n+1)
}
public int[][] pow(int[][] a,int n){//矩阵快速乘,快速幂思想
int [][] ret={{1,0,0},{0,1,0},{0,0,1}};//单位矩阵,初始化作用
while(n>0){
if((n&1)==1){
ret=multiply(ret,a);
}
n>>=1;
a=multiply(a,a);
}
return ret;
}
public int[][] multiply(int [][] a,int [][] othera){//矩阵乘法
// A(m*n) B(n*p) C=AB C(m*p) A的行数 B的列数
int row=a.length,column=othera[0].length;
int [][] ret=new int[row][column];
for(int i=0;i<column;i++){//矩阵行数与列数不一定相同,判断条件不正确
for(int j=0;j<row;j++){
ret[i][j]=addc(a[i],othera,j,column);
}
}
return ret;
}
public int addc(int[] a,int[][] b,int j,int n){//对应行乘对应列,再相加
int sum=0;
for(int k=0;k<n;k++){
sum+=a[k]*b[k][j];
}
return sum;
}
}
此代码中包含矩阵乘法的代码实现,同时包含快速乘的实现。
快速乘迭代实现
快速乘主要思想,即把a的n次方中的n次通过 二进制的形式进行运算。
如2156=2100111000
所以2156=24 * 28* 216 * 2128
迭代实现
public long fastpower2(int a,int n){
System.out.print(fastpower(a,n)+"--\n");
long sum=1;
long ans=a;
while(n!=0){
if(n%2==1){
sum=sum*ans;
}
ans=ans*ans;
n=n/2;
}
return sum;
}
使用long 不适用int 是由于int太过于容易越界,实际上long也容易越界,越界的处理方式还在思考。结合网络资料,可以用数组代替数字类型进行运算。