一、解决的问题:求解一些递推公式的第n项的时候,通过递推公式构造转移矩阵,并用矩阵快速幂可以快速得到第n项的值。特别对于n很大的时候不能循环迭代,只能用矩阵快速幂解决,不然会超时。
二、步骤
1、看是否要化简递推公式(F(n) = F(n-1) + 2F(n-2) + n4的需要化简,具体需不需要化简根据求转移矩阵的决定)
2、得到F(n)和F(n-1)的矩阵序列
3、根据F(n)和F(n-1)的矩阵序列得到中间的转移矩阵
4. 根据转移矩阵编写代码
三、例题
f(n)=af(n-1)+bf(n-2),f(1)=c,f(2)=d。
第一步,不需要化简
第二步,如图
A是转移矩阵,其他两个矩阵系列是去掉系数得到的
第三步,根据矩阵的乘法,可以得A,如图
第四步,如图和例题
例题,求
例题:求第n位斐波那契数mod 10000的大小。其中n的大小高达1000000000
#include<stdio.h>
typedef struct M{ //转移矩阵
int a[2][2]={{1,1},
{1,0}};
}M;
M mMultity(M m1,M m2){ //矩阵相乘
M ans;
for(int i=0;i<2;i++)
for(int j=0;j<2;j++){
ans.a[i][j]=0;
for(int k=0;k<2;k++){
ans.a[i][j]=ans.a[i][j]+m1.a[i][k]*m2.a[k][j];
}
}
return ans;
}
M mpow(M m,int n){ //矩阵的n次方
M ans;
//任何一个矩阵乘于单元矩阵都等于它自己
for(int i=0;i<2;i++)
for(int j=0;j<2;j++){
if(i==j)
ans.a[i][j]=1;
else
ans.a[i][j]=0;
}
while(n!=0){
if(n&1==1) ans=mMultity(m,ans);
m=mMultity(m,m);
n>>=1;
}
return ans;
}
int main(){
int n;
scanf("%d",&n);
M m;
int sum=0;
int a[2][1]={{2},{1}};//1=f(1),2=f(2)的值
if(n>3){
m=mpow(m,n-2);
sum=sum+m.a[0][0]*a[0][0];
sum=sum+m.a[0][1]*a[1][0];
}
else{
sum=a[0][0]+a[1][0];
}
printf("%d",sum);
return 0;
}
三、注意
①、当矩阵A的列数(column)等于矩阵B的行数(row)时,A与B可以相乘。
②、矩阵C的行数等于矩阵A的行数,C的列数等于B的列数。
③、乘积C的第m行第n列的元素等于矩阵A的第m行的元素与矩阵B的第n列对应元素乘积之和。
四、矩阵的乘法
//结构体可以作为返回参数
typedef struct m{
a[4][4];
}M;
M mMutiply(M m1;M m2){
M ans;
for(int i=0;i<4;i++){
for(int j=0;j<4;j++){
ans.a[i][j]=0;
for(int k=0;k<4;k++){
ans.a[i][j]=ans.a[i][j]+m1.a[i][k]*m2.a[k][j];
}
}
}
return ans;
}