众所周知,矩阵具有结合律而不具有交换律,在有的时候,比如让你求斐波那契数列的第10^18项,递推方程是f[i] = f[i - 1] + f[i - 2], f[1] = 1, f[2] = 1, i > 2
正常递推肯定是算不出来的,这时候就要用到矩阵去加速运算。
我们可以设计一个转移矩阵
通过它来快速算出目的项。
前置知识是[[快速幂 & 龟速乘 & 快速乘 | 快速幂]]和[[矩阵]]
核心思想和快速幂一样
只不过这里是用矩阵乘法来实现,所以代码中的乘都变成的矩阵相乘的函数
其限制条件就是, 矩阵快速幂的矩阵A里面是数值必须是个常量, 并且长宽相同
如
0 1
1 1
这就是斐波那契数列的转移矩阵(快速幂加速计算的矩阵)
代码
Mat Mat_qmi(Mat A, int k, int p)
{
Mat res = {A.l, A.r}; // 单位矩阵
for (int i = 1; i <= A.l; i ++ ) res.a[i][i] = 1;
while (k)
{
if (k & 1) res = Mat_mul(res, A, p); // Mat_mul是矩阵乘法实现请见前面"矩阵"那篇笔记
k >>= 1;
A = Mat_mul(A, A, p);
}
return res;
}
单位矩阵
长宽相等, 而且只有行列相等即i == j
的位上为1, 其余为0
如下
1 0 0 0
0 1 0 0
0 0 1 0
0 0 0 1
可以试试这个矩阵无论乘哪个矩阵都是那个矩阵(当然要满足限制条件)
例题
例题[[AcWing 1303. 斐波那契前 n 项和]]
扩展[[AcWing 1304. 佳佳的斐波那契]]