您的位置:首页 > 其它

矩阵快速幂

2015-09-15 21:03 246 查看
本文以矩阵快速幂在求斐波那契数列的第n项

首先我们可以指导斐波那契数列的递推关系 F(n)=F(n-1)+F(n-2)

我们可以把这个关系转为矩阵关系

(F(n),F(n-1))=F(F(n-1),F(n-2))*A

A为某矩阵

我们以数列前三项为例 前三项分别是 1 1 2

(F(3),F(2))=(F(2),F(1))*A

=> (2,1)=(1,1)| 1 1 |

| 1 0 | (我们可以看到,A矩阵就是前面这个矩阵)

(F(4),F(3))=(F(3),F(2))*A

这里面 利用前面的结果

(F(4),F(3))=(1,1)| 1 1 | *A=(1,1)*A^2

| 1 0 |

往后递推下去,我们可以得到这样的递推式

(F(n),F(n-1))=(1,1)*A^(n-2)

这个问题就转换成了,求矩阵A的(n-2)次方。

而求一个数字的n次方,有一个简便方法,比如我们求4的70次方

70的二进制形式是 100 0110

有三个1 分别对应2的一次方,二次方,六次方

所以4^70=(4^64)+(4^4)+(4^2)

所以我们可以先求出来4的一次方,在此基础上乘4,得到4的二次方,然后求三次方,····一直求到4 的6次方,将一次方,二次方,六次方加起来,就可以得到结果。

#include <iostream>
#include <stack>
#include <string>
using namespace std;
//快速矩阵幂
int (*MultipleMatrix(int base[][2],int tmp[][2]))[2]
{
int (*ret)[2]=new int[2][2];
int i,j,k,sum;
for(i=0;i<2;i++)
{
for(j=0;j<2;j++)
{
sum=0;
for(k=0;k<2;k++)
{
sum+=base[i][k]*tmp[k][j];
}
ret[i][j]=sum;
}
}
return ret;

}
int (*nThPowerMatrix(int base[][2],int n))[2]
{
int (*ret)[2]=new int[2][2];
int (*tmp)[2]=base;
ret[0][0]=1;
ret[1][1]=1;
for(int i=0;i<2;i++)
{
for(int j=0;j<2;j++)
{
if(i==j)
ret[i][j]=1;
else
ret[i][j]=0;
}
}
tmp=base;
for(;n>0;n=n>>1)
{
if(n&1)
{
ret=MultipleMatrix(ret,tmp);
}
tmp=MultipleMatrix(base,tmp);
}

return ret;
}

int main()
{

cout<<"以下是使用快速矩阵幂来计算斐波那契数列的第n项值得过程"<<endl;
int n;
cin>>n;
int base[][2]={{1,1},{1,0}};
int (*res)[2]=nThPowerMatrix(base,n-2);
cout<<res[0][0]+res[1][0]<<endl;

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: