Description
Calculate the an % b where a, b and n are all 32bit integers.
Example
For 2^31 % 3 = 2
For 100^1000 % 1000 = 0
看到这题我想到了我之前做过一题pow(x,n).是求x^n.我用了两种方法 。一种是递归,一种是二进制。网址: http://blog.csdn.net/liu981975231/article/details/77503785
这题首先我们用二进制:
解释你可以看我上面的网址,也可以看这一篇:幂模运算。
public int fastPower(int a, int b, int n) {
// write your code here
//利用 (a*c)%b=(c%b * a%b)%b;
if(n==0) return 1;
int result=1;
while(n!=0){
if( (n &1)!=0){
//当b=2147483647时 result有可能超出范围
result=(result%b * a%b)%b;
}
n>>=1;
a*=a;
System.out.println(result);
}
return result%b;
}
说明:利用二进制这样写 会溢出。如 a b n分别为2 ,2147483647,2147483647。则在 result=(result%b * a%b)%b;这一步时,可能会溢出。因为当result相对较大时,而result%2147483647仍为result,然后再乘以a,就会溢出。这个溢出怎么解决呢????我想了很长时间,可惜能力有限没有想出来。所以我换成递归的方法。
public int fastPower(int a, int b, int n) {
// write your code here
if (n == 1) {
return a % b;
} else if (n == 0) {
return 1 % b;
} else if (n < 0) {
return -1;
}
// (a * b) % p = ((a % p) * (b % p)) % p
// use long to prevent overflow
long product = fastPower(a, b, n / 2);
product = (product * product) % b;
//
if (n % 2 == 1) {
product = (product * a) % b;
}
// cast long to int
return (int) product;
}
递归中,需要注意的是要分 n 是奇数还是偶数,奇数的话需要多乘一个 a, 保存乘积值时需要使用long型防止溢出,最后返回时强制转换回int
使用了临时变量product,空间复杂度为 O(1)O(1), 递归层数约为 \log nlogn, 时间复杂度为 O(\log n)O(logn), 栈空间复杂度也为 O(\log n)O(logn).