快速幂
朴素求幂法
最简单的求幂方法,时间复杂度O(n)
int pow(int a,int b) {//a^b
int ans = 1;
for (int i = 1; i <= b; i++) {
ans *= a;
}
return ans;
}
快速幂
时间复杂度为O(logn),假设我们要求a^b
b用二进制表示,b二进制形式的第i位的权为2^(i-1)。例如当b=11,其二进制为
1011,11 = 2³×1 + 2²×0 + 2¹×1 + 2º×1,a11=a(2 ^ 0+2 ^ 1+2 ^ 3)。如此我们按照二进制的权值对b进行计算,逐步获取b的每一位,如果是0,则a的值翻倍;如果是1,则将b乘到结果中,代码如下。
int pow(int a, int b) {
int ans = 1;
while (b != 0) {
if (b % 2 == 1) {
ans *= a;
}
a *= a;
b >>= 1;
}
return ans;
}
注:以上均为b>0的情况,b<0做适当修改即可
快速幂取模
(a * b)%m = ((a%m) * (b%m))%m
由此我们可以推断出
(a^b) % m = (a * a * a…)%c = ((a%c)(a%c)(a%c)*…)%c = (a%c)^b %c;
对快速幂稍作修改即可
int pow_mod(int a, int b,int m) {
int ans = 1;
while (b != 0) {
if (b % 2 == 1) {
ans = (ans * a) % m;
}
a = (a * a) % m;
b >>= 1;
}
return ans;
}