算法 | 快速幂
提升幂运算速度的算法:快速幂
快速幂
复杂度
时间复杂度:
(
代码实现
long long fastpow(long long a, long long b)
{
long long ans = 1;
while (b)
{
if (b % 2 == 1)
ans = ans * a;
a = a * a;
b >>= 1;
}
return ans;
}
算法思路
如果用一般的循环算法,
当然实际情况不会像上面那么顺利,指数不一定能够一直为偶数,所以需要一些额外操作。
以
指数为奇数
指数为偶数
指数为奇数
指数为偶数
指数为偶数
指数为奇数
指数为
在代码中,实际上分出一个底数后,并没有特意将指数减一,而是直接继续操作,原因是之后除二后,整型被截断,一样能得到正确结果,例如上面的 11 >> 1 = 5, 5 >> 1 = 2(整型右移其实和除二一样)
大数取模
一般能用到快速幂算法的,答案都会非常巨大,肯定是超出已有数据类型范围的,所以题目一般会说答案对某个数字取模。快速幂可以完成这个操作,只需要进行很小的修改。
const long long MOD = 20220128;
long long fastpow(long long a, long long b)
{
a %= MOD; // 开头先取一次模
long long ans = 1;
while (b)
{
if (b % 2 == 1)
ans = ans * a % MOD; // 每次运算都取模
a = a * a % MOD; // 每次运算都取模
b >>= 1;
}
return ans;
}
本文采用 CC BY-SA 4.0 许可,本文 Markdown 源码:Haotian-BiJi