快速幂算法

最新推荐文章于 2024-05-21 23:34:41 发布

lero1005

最新推荐文章于 2024-05-21 23:34:41 发布

阅读量159

点赞数

分类专栏：算法学习基础算法

本文链接：https://blog.csdn.net/weixin_41934068/article/details/82192507

版权

算法学习同时被 2 个专栏收录

36 篇文章 1 订阅

订阅专栏

基础算法

13 篇文章 1 订阅

订阅专栏

对于a^n，使用迭代算法的复杂度为O（n），而且cmath的函数库中，pow（a,b）的返回值是int型，稍不注意就会丢失精度。

而快速幂的算法的复杂度则是O（logn）的，原理也很简单，惊了，nice兄dei！

我们知道，任何一个数都可以拆成2的n次幂的和的形式，如 $11 = (1011)_{2} ----->11 = 2^0 + 2^1 + 2^3$ ，那么，对于任意一个数非负数n， $n^{11} = n^{2^{0}+2^{1}+2^{3}} = n^{1+2+8} = n^{1} *(n^{2}) * (n^{8})$ ，我们可以利用一个中间变量base，不停对base进行乘方运算，从而得到n，n^2，n^4等形如 $n^{2^{x}}$ 的数。

那么确定 $n^{2^{x}}$ 该不该乘上呢？很简单，只要把指数表示成二进制，通过&1操作（按位与1，结果是0则最后一位是0，结果是1则最后一位是1）得到指数二进制的最后一位，如果不为0，则可乘一波 $n^{2^{x}}$ ，如果得0则不乘。随后，我们将中间变量base进行乘方运算，再通过>>将指数右移一位，循环即可。

代码奉上：

int qsm(int a, int b)
{
    int ans = 1,base = a;
    while(b != 0)
    {
        if(b & 1)
            ans *= base;
        base *= base;
        b >>= 1;
    }
    return ans;
}