快速幂算法

最新推荐文章于 2021-09-20 21:09:35 发布

MallocLu

最新推荐文章于 2021-09-20 21:09:35 发布

阅读量337

点赞数 1

分类专栏： Algorithm

本文链接：https://blog.csdn.net/qq_42283621/article/details/109641981

版权

Algorithm 专栏收录该内容

10 篇文章 1 订阅

订阅专栏

原文章链接：https://blog.csdn.net/qq_19782019/article/details/85621386

顾名思义，快速幂算法就是用于快速的实现幂运算。

对于a^b（a的b次方）运算，很容易想到可以利用循环实现，但其时间复杂度为O(b)，是耗时比较高的解决办法。

快速幂算法的思想：不断执行指数b减半，底数a平方，直至指数为0。

下面举一个例子：a = 3，b = 10，即3^10

·b减半，a平方，即a=9，b = 5，= 9 ^ 5；

·b减半，a平方，此时b为奇数，可以拆分为4+1再执行，即 = 9 ^ (4 + 1) = (9 ^ 4) * (9 ^ 1) = (81 ^ 2) * (9 ^ 1)；

·b减半，a平方，即a=81*81=6561，b = 2/1=1，= (6561 ^ 1) * (9 ^ 1)；

·b减半，a平方，此时b为奇数，可以拆分为0+1再执行，即 = (6561 ^ (0 + 1)) * (9 ^ 1) = (6561 * 0) * (6561 ^ 1) * (9 ^ 1)；

·b等于0，算法结束。

发现：最后3 ^ 10 = (6561 * 0) * (6561 ^ 1) * (9 ^ 1)，其有两个因子，即6561和9，这两个因子从何处而来呢？可以发现，每次当指数b是奇数时，应当写成(b-1)+1的形式，此时的底数a即为最终结果的一个乘法因子。

总结：当指数为偶数时，底数平方，指数减半；

当指数为奇数时，最终结果添加乘法因子a（当前状态下的底数），然后底数平方，指数减1后减半；

算法实现：

long long fastPower(long long base, long long power) 
{
    long long result = 1;
    while (power > 0) 
    {
        if (power % 2 == 0) 
        {
            power /= 2;
            base *= base;
        } 
        else
        {
            result *= base;
            power = (power - 1) / 2;
            base *= base;
        }
    }
    return result;
}

代码简化：

·对于整数power，当其为奇数时，(power - 1) / 2和power / 2相同，所有else块中可以改成power /= 2;

·分支块中存在公共部分power /= 2，base *= base，可以提出；

long long fastPower(long long base, long long power) 
{
    long long result = 1;
    while (power > 0) 
    {
        if (power % 2 != 0) 
            result *= base;
        power /= 2;
        base *= base;
    }
    return result;
}

最终优化：将power % 2 != 0和power /= 2改为更底层、速度更快的位运算

long long fastPower(long long base, long long power) 
{
    long long result = 1;
    while (power > 0) 
    {
        if (power & 1) 
            result *= base;
        power >>= 1;
        base *= base;
    }
    return result;
}