1. 递归写法的快速幂
给定三个正整数a、b、m(a < 109,b < 1018,1 < m < 109)求ab % m
快速幂基于以下事实:
- 如果b是奇数,那么有ab = a * ab-1
- 如果b是偶数,那么有ab = ab/2 * ab/2
显然b为奇数的情况总是可以转换为b为偶数的情况,而b为偶数的情况又可以进一步转换为b/2的情况。经过log(b)级别的转换后,就可以把b变为0,而任何数的0次幂都为1。
快速幂的递归写法,代码如下:
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <cstring>
#include <queue>
using namespace std;
// 快速幂的递归写法
typedef long long LL;
LL binaryPow(LL a, LL b, LL m)
{
// 递归边界
if (b == 0) return 1;
// b为奇数时
if (b & 1) return a * binaryPow(a, b-1, m) % m;
else
{
LL mul = binaryPow(a, b/2, m);
return mul * mul % m;
}
}
int main()
{
LL res = binaryPow(2, 10, 1000);
printf("%lld", res); // 24
return 0;
}
注:
- 代码中判断条件(b & 1)的意思为位与操作,判断b的末位是否为1
- 如果初始条件有a > m的情况,则需要在进入函数前a 对 m取模(???)。
- 如果m为1,可以直接在函数外部特判为0(任何数对1取模都为0)。
2. 迭代写法的快速幂
算法描述如下:
- 令ans = 1,存放累积的结果
- 判断b的二进制末尾是否为1,如果是,则令ans *= a,转3;如果不是,则转4
- 令a = a * a,转4
- b右移
(可类比2进制的求值方式)
快速幂的迭代写法,代码如下:
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <cstring>
#include <queue>
using namespace std;
// 快速幂的迭代写法
typedef long long LL;
LL binaryPow(LL a, LL b, LL m)
{
LL ans = 1;
while (b > 0)
{
// 判断最后一位是否为奇数
if (b & 1)
{
ans *= a % m;
}
a = a * a % m;
b >>= 1; // 将b的二进制右移一位,即b = b >> 1 或b = b / 2
}
return ans;
}
int main()
{
LL res = binaryPow(2, 3, 50000);
printf("%lld\n", res);
return 0;
}
注:
- 代码中的 >> 运算为,将该数的二进制形式右移
- 实际应用中,迭代写法与递归写法效率相差不大