求a的b次方对p取模的值,其中1≤a,b,p≤10^9
根据数学常识,每一个二进制数可以唯一表示为若干指数不重复的2的次幂的和
因此b可以根据这个数学常识表示为若干指数不重复的2的次幂的和
C++代码
int power(int a, int b, int p){
int ans = 1 % p; //这里取模是为了处理模为0的情况
while(b){
if(b & 1){
ans = ans * a % p;
}
a = a * a % p;
b >>= 1;
}
return ans;
}
Python代码
def power(a, b, p):
ans = 1 % p
while b:
if b & 1:
ans = ans * a % p
a = a * a % p
b >>= 1
return ans
Java代码
public static int power(int a, int b, int p){
int ans = 1 % p;
while (b != 0){
if ((b & 1) == 1){
ans = ans * a % p;
}
a = a * a % p;
b = b >> 1;
}
return ans;
}