快速幂就是快速算底数的n次幂。其时间复杂度为 O(log₂N), 与朴素的O(N)相比效率有了极大的提高。
代码实现:
代码语言:javascript复制int pow_mod(int a, int k)
{
int ans = 1;
while(k)
{
if(k % 2)
ans *= a;
a *= a;
k /= 2;
}
return ans;
}
原理我理解的就是把指数换成二进制进行运算,如果二进制位数的数是1,那么让我们要的结果ans乘底数a,然后a要自乘一次,如果是二进制位数的数是0,那么ans不进行操作,a还是要自乘一次的,k就是指数,比如2^6,这里a就是2,k就是6。
对于快速幂取模可以利用a * b % m = [(a % m) * (b % m)] % m;
代码实现:
代码语言:javascript复制long long int pow_mod(long long int a, long long int k)
{
long long int ans = 1;
a %= MOD;
while(k)
{
if(k % 2) ans *= a;
a = (a * a) % MOD;
k /= 2;
ans %= MOD;
}
return ans;
}