快速幂算法

转载自：https://wenku.baidu.com/view/2384ecc02cc58bd63186bdf6.html

我们先从简单的例子入手:求a^b mod c =？

算法1:首先直接地来设计这个算法:

int ans = 1;
for (int i = 1;i <= b; i++)
{
   ans = ans * a;
}
ans = ans % c; 

 这个算法的时间复杂度体现在for循环中，为O(b).这个算法存在着明显的问题，如果a和b过大，很容易溢出。

 那么，我们先来看第一个改进方案:在讲这个改进方案之前，要先有这样一个公式:

                                                        a^b mod c = (a mod c)^b mod c 

这个公式大家在离散数学或数论中应当学过，不过这里为了方便大家阅读，还是给出证明:

引理1:

公式: (ab) mod c= [(a mod c)*(b mod c)] mod c

证明: a mod c = d =>a= tc+d;

b mod c = e =>b= kc + e

ab mod c = (tc+d)(kc+e) mod c

                =(tkc^2 + (te + dk)c + de) mod c

                =de mod c =  [(a mod c)*(b mod c)] mod c

上面的公式为下面公式的引理，即积的取余等于取余的积的取余

公式: a^b mod c = (a mod c)^b mod c

证明: (a mod c)^b mod c

 =[((a mod c) mod c)^b] mod c

(a mod c)^b mod c = a^b mod c

证明了以上的公式以后，我们可以先让a关于c取余，这样可以大大减少a的大小，于是进行以下改进。

算法2:

int ans = 1;
a = a % c; //加上这一句
for(int i = 1;i <= b;i++)
{
   ans= ans * a;
}
ans = ans % c;

聪明的读者应该可以想到，既然某个因子取余之后相乘再取余保持余数不变，那么新算得的ans也可以进行取余，所以得到比较良好的改进版本。

算法3：

int ans = 1;
a = a % c;
for(int i = 1;i <= b;i++)
{
   ans= (ans * a) % c; //这里再取了一次余
}
ans = ans % c;

这个算法时间复杂度没有改进，仍为O(b)，不过已经好很多了，但是在c过大的条件下，还是有可能超时，所以，我们推出以下的快速幂算法.

快速幂算法依赖于以下明显的公式，我就不证明了。

a^b mod c = ((a^b)^b/2) mod c. b是偶数

a^b mod c = ((a^b)^b/2 * a) mod c. b是奇数

有了以上两个公式后，我们可以得出以下结论:

1.如果b是偶数，我们可以记k = a^2 mod c 那么求k^b/2 mod c 就可以了

2.如果b是奇数，我们可以记k = a^2 mod c 那么求(k^b/2 mod c * a) mod c 就可以了

那么我们可以得到以下算法:

算法4:

int ans = 1;
a = a % c;
if (b % 2 == 1)
   ans = (ans * a) % c; //如果b是奇数，要多求一步，可以提前算到 ans 中
k = (a * a) % c;
for(int i = 1;i <= b/2;i++)
{
   ans = (ans * k) % c;
}
ans = ans % c;

我们可以看到，我们把时间复杂度变成了O(b/2)。当然，这样子治标不治本。但我们可以看到，当我们令 k = (a * a) mod c 时，状态已经发生了变化，我们所要求的最终结果即为k^b/2 mod c，而不是原来的a^b mod c,所以我们发现这个过程是可以迭代下去的。当然，对于奇数的情形会多出一项 a mod c 所以为了完成迭代，当b是奇数时，我们通过ans = (ans * a) % c，来弥补多出来的这一项，此时剩余的部分就可以进行迭代了。

形如上式的迭代下去后，当b = 0 时，所有的因子都已经相乘，算法结束。于是便可以在O(logb)的时间内完成。于是，有了最终算法: 快速幂算法。

算法5:快速幂算法

int ans = 1;
a = a % c;
while(b>0)
{
    if(b % 2 == 1)
       ans = (ans * a) % c;
    b = b/2;
    a = a * a % c;
}

将上述代码写成函数:

typedef long long ll;
ll quick_pow(ll a,ll b,ll c)
{
    ll ans = 1;
    a = a % c;
    while(b>0)
    {
      if(b % 2 == 1)
         ans = (ans * a) % c;
      b = b/2;
      a = (a * a) % c;
    }
  return ans;
}