int ans = 1;
for(int i = 0; i < b; i++)
ans = ans * a;
ans = ans % c;
这个算法不仅复杂度是O(b)" role="presentation">O(b),而且一旦a" role="presentation">a或者b" role="presentation">b的值过大,就会发生溢出。下面,我们来看一个新的算法:
我们来证明ab%c=(a%c)b%c" role="presentation">ab%c=(a%c)b%c
设a=k×c+e" role="presentation">a=k×c+e,带入上式得(k×c+e)b%c=eb%c=(a%c)b%c" role="presentation">(k×c+e)b%c=eb%c=(a%c)b%c
证毕。
利用这个公式,我们可以改进刚才的算法
int ans = 1;
a = a % c;
for(int i = 0; i < b; i++)
ans = ans * a;
ans = ans % c;
这个算法的时间复杂度仍然是O(b)" role="presentation">O(b),还是会在b" role="presentation">b很大的时候发生溢出。我们还需要继续改进算法。
继续改进
我们的最终算法依赖下面这两个公式:
1. 当b" role="presentation">b是偶数时,ab%c=((a2)b/2)%c" role="presentation">ab%c=((a2)b/2)%c
2. 当b" role="presentation">b是奇数时,ab%c=((a2)b/2×a)%c" role="presentation">ab%c=((a2)b/2×a)%c
这两个公式是很容易证明的,我们不在赘述。
根据这两个公式,我们有:
1. 当b" role="presentation">b是偶数时,ab%c=(a2)b/2%c=(a2%c)b/2%c" role="presentation">ab%c=(a2)b/2%c=(a2%c)b/2%c
2. 当b" role="presentation">b是奇数时,ab%c=((a2)b/2×a)%c=[(a2%c)b/2%c×(a%c)]%c" role="presentation">ab%c=((a2)b/2×a)%c=[(a2%c)b/2%c×(a%c)]%c
这样就可以写出快速幂取模的递归算法了
#include int PowerMod(int a, int b, int c){
if(b == 1)
return a % c;
else{
if(b % 2 == 0)
return PowerMod(a * a % c, b / 2, c);
elsereturn PowerMod(a * a % c, b / 2, c) * (a % c) % c;
}
}
int main()
{
printf("%d
", PowerMod(2, 8, 3));
return0;
}
这时我们的算法不但不会溢出,而且时间复杂度也降到了log2(b)" role="presentation">log2(b)