快速幂(Exponentiation by Squaring)
快速幂是一种高效计算大整数幂次的方法,其时间复杂度为O(logn) ,适用于计算 an%m之类的运算。
快速幂的基本原理
快速幂基于以下观察:要计算一个数的幂次,如果我们知道 an 的结果,我们可以很容易地计算 a2n 的结果。具体来说:
- an = an/2 * an/2 如果n是偶数
- an = a(n-1)/2 * a(n-1)/2 * a 如果n是奇数
因此,通过不断地将指数n减半,我们可以快速地计算出最终的结果。
算法步骤
- 初始条件: 给定基数 a、指数 n、模 m(如果需要),初始化结果 (res = 1)。
- 循环过程:
- 如果 n 是奇数,更新结果:res = (res * a) % m。
- 更新基数:a = (a * a) % m。
- 将指数 n 右移一位(相当于 n = n / 2)。
- 终止条件: 当 n = 0 时,循环结束,返回结果。
快速幂的代码实现
private int fastPow(int base, int exp, int mod) {
int result = 1;
base = base % mod; // 确保 base 小于 mod
while (exp > 0) {
// 如果 exp 是奇数,乘以当前的 base
if ((exp & 1) == 1) {
result = (result * base) % mod;
}
// 将 base 平方,并更新 exp
base = (base * base) % mod;
exp >>= 1; // exp = exp / 2
}
return result;
}
解释
- 初始化:
result
为最终结果,开始时为 1。base
和mod
初始化确保我们计算的是 (base % mod) 的幂。 - 主循环:
- 如果
exp
是奇数(exp & 1
为 1),则将base
乘以result
。 - 将
base
平方,并且指数exp
减半(右移一位)。
- 如果
- 终止: 最后
result
就是我们所求的 an % m。
这种方法通过减少指数的指数级增长有效地提高了运算速度,特别适合大数运算中对性能有要求的场景。