Java教程

快速幂算法

本文主要是介绍快速幂算法,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

我们首先来看一个问题:

给定三个正整数a,b,m(a<10^9, b<10^18, 1<m<10^9),求 a^b%m。

如果用循环来写,不断乘上a再取模,时间复杂度为O(b)。很容易超时。

所以我们考虑快速幂算法。它基于二分的思想,也被称为二分幂

递归写法

1.如果b是奇数,那么有:\(a^b = a*a^{b-1}\)

2.如果b是偶数,那么有:\(a^b = a^{b/2}*a^{b/2}\)

typedef long long LL;
// 递归求a^b%m
LL binaryPow(LL a,LL b,LL m){
    if (b == 0) return 1;

    if (b%2 == 1) return a * binaryPow(a,b-1,m) % m;
    else{
        LL mul = binaryPow(a,b >>1,m);
        return mul * mul % m;
    }
}
  • 如果初始时a>=m,需要执行函数时先对a%m。
  • 如果m==1,直接在函数外部特判为0。

迭代写法

把b写成二进制的形式,那么b可以写成若干二次幂之和。

例如13的二进制形式为1101, \(13=2^3+2^2+2^0,a^{13}=a^8\ *a^4\ *a^1\) 。

不难推出:当b的二进制的第i(从0开始)位是1时,初值ans=1要乘上\(a^{2{^i}}\)。由于我们每次迭代时令a平方

(\(a=a^{2i}\)),所以枚举当前第i位时,a已经迭代到初始a的\(2^i\)次方,如果是1,就令ans乘上a。

typedef long long LL;
//  迭代求a^b%m
LL binaryPow(LL a,LL b,LL m){
    LL ans = 1;
    while (b > 0){
        // 等价于b%2 == 1
        if (b&1) ans = ans*a%m;
        a = a*a%m;
        b >>1;
    }
    return ans;
}

在实际应用场景,两种写法效率差不多。

大数取模公式总结:

参考资料1:博客园

参考资料2:《算法笔记》

这篇关于快速幂算法的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!