矩阵快速幂_矩阵快速幂优化-CSDN博客

无耻的粘自Margatroid

前置技能

矩阵乘法

复杂度为 $O(n^3)$ ，有复杂度稍低的分治写法，不过意义不大（毕竟你的矩阵这么小）

$A,B$ 是两个矩阵，其中 $A$ 是 $m\times n$ 的矩阵， $B$ 是 $x\times y$ 的矩阵

A = [a 11 a 21 a 12 a 22 a 13 a 23], B = ⎡ ⎣ ⎢ b 11 b 21 b 31 b 12 b 22 b 32 ⎤ ⎦ ⎥

$A= \left[ \begin{array}{} a_{11}&a_{12}&a_{13}\\ a_{21}&a_{22}&a_{23}\\ \end{array} \right] ,\ B= \left[ \begin{array}{} b_{11}&b_{12}\\ b_{21}&b_{22}\\ b_{31}&b_{32} \end{array} \right]$
当且仅当

n=x $n=x$ 时

A⋅B $A\cdot B$ 有意义。

注意：矩阵乘法不满足交换律，设 $C=A\cdot B$ ，则

c i k = \sum j = 1 n a i j b j k

$c_{ik}=\sum_{j=1}^na_{ij}b_{jk}$
一般的，我们用矩阵去左乘一个向量，来对该向量进行 线性变换。

如果对于矩阵乘法仅仅是会写代码的程度，请深入理解并能够手算，否则不要阅读以下内容。

快速幂

复杂度 $O(\log n)$

这个没啥好说的，如果不会可以上网搜或者看我之前的博客

矩阵快速幂

矩阵加上快速幂，可以迅速求出线性递推式的第 $n$ 项，或者是前 $n$ 项和（必须是线性组合）

求斐波那契数列

斐波那契数列的定义：

F 1 = F 2 = 1, F n = F n - 1 + F n - 2

$F_1=F_2=1,\ F_n=F_{n-1}+F_{n-2}$

我们注意到，该数列的第 $n$ 项是第 $n-1$ 项和第 $n-2$ 项的线性组合（~~你tm在放屁~~）

我们构造一个列向量：

A = [F n - 1 F n - 2]

$A=\left[ \begin{array}{cc|c} F_{n-1}\\ F_{n-2} \end{array} \right]$

然后我们考虑第 $n+1$ 项，是第 $n$ 项和第 $n-1$ 项的线性组合（~~简直是废话~~）

同样的，我们也可以构造一个列向量：

B = [F n - 1 + F n - 2 F n - 1]

$B=\left[ \begin{array}{cc|c} F_{n-1}+F_{n-2}\\ F_{n-1} \end{array} \right]$

然后我们考虑一下怎么把 $A$ 经过一波骚变换而把它变成 $B$ ，由于这是向量之间线性组合，左乘一个矩阵是可以的：

[1011] [F n - 1 F n - 2] = [F n - 1 + F n - 2 F n - 1]

$\left[ \begin{array}{cc|c} 1&1\\ 0&1 \end{array} \right] \left[ \begin{array}{cc|c} F_{n-1}\\ F_{n-2} \end{array} \right] = \left[ \begin{array}{cc|c} F_{n-1}+F_{n-2}\\ F_{n-1} \end{array} \right]$

嗯，就这样变换。

如果我要求第 $n$ 项，就再左乘 $n-2$ 个同样的矩阵，所以，我们有：

[1011] n - 2 [F 1 F 2] = [F n F n - 1]

$\left[ \begin{array}{} 1&1\\ 0&1 \end{array} \right]^{n-2} \left[ \begin{array}{} F_1\\ F_2 \end{array} \right] = \left[ \begin{array}{} F_n\\ F_{n-1} \end{array} \right]$
然后左边的那个矩阵的幂可以用快速幂处理，可以在

O(logn) $O(\log n)$ 时间内求出，再乘一下那个列向量，就得到了我们要求的东西。

进阶版：求两项的线性递推式

题目：COGS

其实就是给斐波那契数列加上了系数。

矩阵会变成这样：

[a 0 b 1] [F n - 1 F n - 2] = [a F n - 1 + b F n - 2 F n - 1]

$\left[ \begin{array}{cc|c} a&b\\ 0&1 \end{array} \right] \left[ \begin{array}{cc|c} F_{n-1}\\ F_{n-2} \end{array} \right] = \left[ \begin{array}{cc|c} aF_{n-1}+bF_{n-2}\\ F_{n-1} \end{array} \right]$

极限版：求n项的线性递推式

题目：COGS

嗯。。。这个就比较毒瘤了。。。

还是要构造矩阵。。（怕是要敲公式敲死）

还是去网上搜一个矩阵粘下来吧（懒癌爆发）

好像并没有人手敲矩阵。。决定吃螃蟹。。。

求

f (n) = a 1 f (n - 1) + a 2 f (n - 2) + . . . + a d f (n - d) (n > d)

$f(n)=a_1f(n-1)+a_2f(n-2)+...+a_df(n-d)\ (n>d)$

其中 $a_1,a_2,\cdots a_n$ 和 $f(1),f(2),\cdots f(n)$ 已知。

大力构造矩阵：（md真丑）

⎡ ⎣ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ a 1 100 ⋮ 0 a 2 010 ⋮ 0 a 3 001 ⋮ 0 \dots \dots \dots \dots ⋱ \dots 1 a n 000 ⋮ 0 ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎡ ⎣ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ f (n) f (n - 1) f (n - 2) f (n - 3) ⋮ ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ = ⎡ ⎣ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ f (n + 1) f (n) f (n - 1) f (n - 2) ⋮ ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥

$\left[ \begin{array}{cc} a_1&a_2&a_3&\cdots&&&a_n\\ 1&0&0&\cdots&&&0\\ 0&1&0&\cdots&&&0\\ 0&0&1&\cdots&&&0\\ \vdots&\vdots&\vdots&\ddots&&&\vdots\\ 0&0&0&\cdots&&1&0 \end{array} \right] \left[ \begin{array}{cc|c} f(n)\\ f(n-1)\\ f(n-2)\\ f(n-3)\\ \vdots \end{array} \right] = \left[ \begin{array}{cc|c} f(n+1)\\ f(n)\\ f(n-1)\\ f(n-2)\\ \vdots \end{array} \right]$
啊。。我尽力了。。。就是这样。。敲公式真累。。。

求斐波那契数列的前n项和

即求

\sum i = 1 n F i

$\sum_{i=1}^nF_i$
这个和之前的还是不太一样的，没有明确的递推式。

我们注意到前 $n$ 项和 $sum_n$ 是 $sum_{n-1}$ 和 $F_n$ 的线性组合，然后构造矩阵：

⎡ ⎣ ⎢ 100111110 ⎤ ⎦ ⎥ ⎡ ⎣ ⎢ s u m n - 1 F n - 1 F n - 2 ⎤ ⎦ ⎥ = ⎡ ⎣ ⎢ s u m n F n F n + 1 ⎤ ⎦ ⎥

$\left[ \begin{array}{ccc} 1&1&1\\ 0&1&1\\ 0&1&0 \end{array} \right] \left[ \begin{array}{cc} sum_{n-1}\\ F_{n-1}\\ F_{n-2} \end{array} \right] = \left[\begin{array}{cc} sum_{n}\\ F_n\\ F_{n+1} \end{array} \right]$

求自然数列的前n项平方和

即求：

\sum i = 1 n i 2

$\sum_{i=1}^ni^2$
由于

sumn+1 $sum_{n+1}$ 是

sumn $sum_n$ 与

(i+1)2 $(i+1)^2$ 的线性组合，

(i+1)2 $(i+1)^2$ 是

i2, i, 1 $i^2,\ i,\ 1$ 的线性组合，因此可以构造矩阵：

⎡ ⎣ ⎢ ⎢ ⎢ ⎢ 1000110022101111 ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ ⎡ ⎣ ⎢ ⎢ ⎢ ⎢ s u m n i 2 i 1 ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ = ⎡ ⎣ ⎢ ⎢ ⎢ ⎢ ⎢ s u m n + 1 (i + 1) 2 (i + 1) 1 ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ ⎥

$\left[ \begin{array}{ccccc|c} 1&1&2&1\\ 0&1&2&1\\ 0&0&1&1\\ 0&0&0&1\\ \end{array} \right] \left[ \begin{array}{ccccc|c} sum_n\\ i^2\\ i\\ 1\\ \end{array} \right] = \left[ \begin{array}{ccccc|c} sum_{n+1}\\ (i+1)^2\\ (i+1)\\ 1 \end{array} \right]$

求斐波那契数列的前n项平方和

即求

\sum i = 1 n F 2 i

$\sum_{i=1}^nF_i^2$
注意到

sumn $sum_n$ 是

F2n $F_n^2$ 与

sumn−1 $sum_{n-1}$ 的线性组合，而

F2n $F_n^2$ 是

F2n−1 $F_{n-1}^2$ 和

F2n−2 $F_{n-2}^2$ 和

Fn−1Fn−2 $F_{n-1}F_{n-2}$ 的线性组合，所以构造矩阵：

⎡ ⎣ ⎢ ⎢ ⎢ ⎢ 1000111111002201 ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ \cdot ⎡ ⎣ ⎢ ⎢ ⎢ ⎢ ⎢ s u m n - 1 F 2 n - 1 F 2 n - 2 F n - 1 \cdot F n - 2 ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ ⎥ = ⎡ ⎣ ⎢ ⎢ ⎢ ⎢ ⎢ s u m n F 2 n F 2 n - 1 F n \cdot F n - 1 ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ ⎥

$\left[ \begin{array}{cccc|c} 1&1&1&2\\ 0&1&1&2\\ 0&1&0&0\\ 0&1&0&1\\ \end{array} \right] \cdot \left[ \begin{array}{cccc|c} sum_{n-1}\\ F_{n-1}^2\\ F_{n-2}^2\\ F_{n-1}\cdot F_{n-2} \end{array} \right] = \left[ \begin{array}{cccc|c} sum_{n}\\ F_n^2\\ F_{n-1}^2\\ F_n\cdot F_{n-1} \end{array} \right]$

注意在推矩阵的最后以后时使用了一些骚操作，可以手推一下验证一下正确性

求自然数列的前n项立方和

即求：

\sum i = 1 n i 3

$\sum_{i=1}^ni^3$
注意到

sumn+1 $sum_{n+1}$ 是

(i+1)3 $(i+1)^3$ 与

sumn $sum_{n}$ 的线性组合，而

(i+1)3=i3+3i2+3i+1 $(i+1)^3=i^3+3i^2+3i+1$ ，是

i3,i2,i,1 $i^3,i^2,i,1$ 的线性组合，

(i+1)2=i2+2i+1 $(i+1)^2=i^2+2i+1$ ，是

i2 $i^2$ 与

i $i$ 的线性组合，

i+1 $i+1$ 是

i $i$ 与

1 $1$ 的线性组合，因此构造矩阵：

⎡ ⎣ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ 1000011000331003321011111 ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎡ ⎣ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ s u m n i 3 i 2 i 1 ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ = ⎡ ⎣ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ ⎢ s u m n + 1 (i + 1) 3 (i + 1) 2 (i + 1) 1 ⎤ ⎦ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥ ⎥

$\left[ \begin{array}{ccccc|c} 1&1&3&3&1\\ 0&1&3&3&1\\ 0&0&1&2&1\\ 0&0&0&1&1\\ 0&0&0&0&1\\ \end{array} \right] \left[ \begin{array}{ccccc|c} sum_n\\ i^3\\ i^2\\ i\\ 1\\ \end{array} \right] = \left[ \begin{array}{ccccc|c} sum_{n+1}\\ (i+1)^3\\ (i+1)^2\\ (i+1)\\ 1 \end{array} \right]$