快速幂与矩阵快速幂

一、快速幂

1.背景

快速幂其实是一种用于快速计算出$x^n$的算法，其中指数n比较大。

2.对比

• 传统求幂算法：时间复杂度为O(指数n)
• 快速幂算法：核心思想是每一步都把指数分成两半，而相应的底数做平方运算，这样不仅能把非常大的指数给不断变小，所需要执行的循环次数也在变小，而最后的结果却一直不会变。

3.快速幂的理解

①从二进制的角度进行理解

对于任意的十进制整数n，假设其二进制表示为“$b_m...b_3{b}_2{b}_1$”，则有二进制转十进制：$$n=2^0{b}_1+2^1{b}_2+2^2{b}_3+...+2^{m-1}{b}_m$$
从而有x的二进制展开为：$$x^n=x^{2^0{b}_1+2^1{b}_2+2^2{b}_3+...+2^{m-1}{b}_m}=x^{2^0{b}_1}\cdot x^{2^1{b}_2}\cdot x^{2^2{b}_3}\cdot ...\cdot x^{2^{m-1}{b}_m}$$
根据上述的推导，可以把计算$x^n$转化成解决以下两个问题：

• 计算$x^{2^0},x^{2^1},x^{2^2},...,x^{2^{m-1}}$:循环赋值$x=x^2$即可
• 获取二进制各位$b_1,b_2,b_3,...,b_m$的值：循环执行如下的操作

n&1(与操作)：判断二进制位的最后一位是否为1
n>>1(移位操作)：n右移一位（可理解为删除二进制位的最后一位）

因此，应用以上操作，可在循环中依次计算$x^{2^0},x^{2^1},x^{2^2},...,x^{2^{m-1}}$的值，并将这些值相乘即可。

②从二分法的角度进行理解

快速幂实际上是二分思想的一种应用。
二分推导：$x^n=x^{n/2}$ · $x^{n/2}$，若要令$n/2$为整数，则需要分n为奇数和偶数两种情况：

• 当n为偶数：$x^n=(x^2{)}^{n//2}$
• 当n为奇数：$x^n=x\cdot (x^2{)}^{n//2}$，会比多出一项x

根据二分推导，可以通过循环$x=x^2$的操作，每次把幂从n降为n//2，直至将幂降为0。

设sum=1，设初始状态$x^n=x^n\cdot sum$。在循环二分的时候，当n为偶数，底数由$x$变为$x^2$，指数$n$变为$n//2$；当n是奇数时，$sum=sum\ast x$底数由$x$变为$x^2$，指数$n$变为$n//2$。

算法执行流程如下：

• 当x==0时：直接返回0
• 初始化sum=1
• 当n < 0：把问题转化为n >= 0的范围内，即x = 1/x，n = -n
• 循环计算（当n == 0结束循环）：当n&1 == 1，则sum *= x —>x = x^2 —>n >> 1

二、矩阵快速幂

矩阵快速幂实际上就是将快速幂算法里的数字换成矩阵，

矩阵快速幂算法 = 矩阵乘法 + 快速幂算法

应用实例1：斐波那契数列—$f(n)=f(n-1)+f(n-2)$
$$\begin{array}{c}\left[\begin{array}{c}f(n)\\ f(n-1)\end{array}\right]=\left[\begin{array}{cc}1& 1\\ 1& 0\end{array}\right]\left[\begin{array}{c}f(n-1)\\ f(n-2)\end{array}\right]=...={\left[\begin{array}{cc}1& 1\\ 1& 0\end{array}\right]}^{n-1}\left[\begin{array}{c}f(1)\\ f(0)\end{array}\right]\end{array}$$

应用实例2：$f(n)=af(n-1)+bf(n-2)+c$
$$\begin{array}{c}\left[\begin{array}{c}f(n)\\ f(n-1)\\ c\end{array}\right]=\left[\begin{array}{ccc}a& b& 1\\ 1& 0& 0\\ 0& 0& 1\end{array}\right]\left[\begin{array}{c}f(n-1)\\ f(n-2)\\ c\end{array}\right]=...={\left[\begin{array}{ccc}a& b& 1\\ 1& 0& 0\\ 0& 0& 1\end{array}\right]}^{n-1}\left[\begin{array}{c}f(1)\\ f(0)\\ c\end{array}\right]\end{array}$$