快速幂原理及伪代码

最新推荐文章于 2022-03-06 01:45:00 发布

SmileTAT

最新推荐文章于 2022-03-06 01:45:00 发布

阅读量666

点赞数

分类专栏： LeetCode 文章标签：矩阵线性代数算法

本文链接：https://blog.csdn.net/sxl_smile/article/details/118435262

版权

LeetCode 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

应用场景是求解 $M^{n}$ 形式的连续幂次元算， $M$ 可以是数值或矩阵形式。经典算法题为斐波那契数列问题。

快速幂的核心思想是将 $n$ 以二进制方式表示成 $n=\sum_{i=0}^{\left \lceil \log_{2}{n}\right \rceil} a_{i}*2^{i}$ ，则所要求解的原式可以表示为 $M^{\sum_{i=0}^{\left \lceil \log_{2}{n}\right \rceil} a_{i}*2^{i}} = \prod_{i=0}^{\left \lceil \log_{2}{n}\right \rceil} M^{a_{i}*2^{i}}$ ， $a_{i}$ 取值为0或1取决于二进制表示的n在 $i$ 位的值， $a_{i} = (n>>i) \&1$ 。 $i=0,base=M; i=1, base=M^2; i=2, base=M^4,......$ 。可以看到随着 $i$ 增加1则 $base$ 为上一个的平方。亦可从公式推导 $\frac{M^{2^{i+1}}}{M^{2^i}} = M^{2^i}$ 。

按上述原理可写成如下的代码，其中multiply是所有可乘的方法的抽象表示，例如数值乘，矩阵乘。

时间复杂度： $O(\log n)$

空间复杂度： $O(1)$

result = 1; // 或I单位矩阵
base = M;
while (n > 0) {
  if (n & 1) { // a_i = 1
    result = multiply(result, base);
  }
  base = multiply(base, base);
  n >> 1;
}

SmileTAT

关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
快速幂原理及伪代码

应用场景是求解形式的连续幂次元算，可以是数值或矩阵形式。经典算法题为斐波那契数列问题。快速幂的核心思想是将以二进制方式表示成，则所要求解的原式可以表示为，取值为0或1取决于二进制表示的n在位的值，。。可以看到随着增加1则为上一个的平方。亦可从公式推导。按上述原理可写成如下的代码，其中multiply是所有可乘的方法的抽象表示，例如数值乘，矩阵乘。时间复杂度：空间复杂度：result = 1; // 或I单位矩阵base = M;while (n > 0) { if (n
复制链接

扫一扫