快速幂取模算法证明及实现

2019-04-13 12:41发布

快速幂取模

快速幂取模就是使用更快的方式计算幂式的模,例如当a" role="presentation">ab" role="presentation">b很大时,用传统的方法计算ab%c" role="presentation">ab%c会发生溢出,这时我们就需要使用快速幂取模算法。

传统算法计算ab%c" role="presentation">ab%c

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的算法

我们来证明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); else return PowerMod(a * a % c, b / 2, c) * (a % c) % c; } } int main() { printf("%d ", PowerMod(2, 8, 3)); return 0; } 这时我们的算法不但不会溢出,而且时间复杂度也降到了log2(b)" role="presentation">log2(b)