对于式子a^b%p,如果直接先计算a^b,再取余,算法复杂度是非常高的,因此需要使用快速幂算法对其进行优化。
算法原理如下:
首先我们先利用第三个式子,写出快速乘法取余算法:
long long quick_mul(long long a, long long b, long long p)
{
long long ans;
ans = ((a%p)*(b%p)) % p;
return ans;
}
接下来就可以结合第三第四个式子进行快速幂取余算法了
我们对其优化的方向很明确,就是尽量减少for循环次数,
1.首先,我们发现a^b=(a^2)^(b/2),这样就只需循环b/2次,但复杂度仍很高
2.接下来就对第一步进行迭代,不断地对a在平方的基础上再取平方,对b除以2,这样迭代后的效果便是只需循环次便可,极大减少了复杂度
3.同时细节部分便是对于b除2后为奇数的情况则需额外进行ans=ans*a%p操作
接下来是代码部分:
long long quick_pow(long long a, long long b, long long p)
{
long long ans=1;
a = a % p;
while (b)
{
if(b&1) ans = quick_mul(ans,a,p);
a = quick_mul(a,a,p);
b >>= 1;
}
return ans;
}